diff --git a/src/controllers/render_controller.py b/src/controllers/render_controller.py index d362dd50cb8aec912d358e892164004c56f791b1..92cc236c09e61d0a34ae732f46803a29180daa09 100644 --- a/src/controllers/render_controller.py +++ b/src/controllers/render_controller.py @@ -2,5 +2,17 @@ import os from gradio_client import Client, handle_file class RenderController: - def get_render(self, hf_token='') -> str: + def get_render(self, hf_token='', text="Hello!!", + selected_model="Model 1 (Turbo Realism)") -> str: + + client = Client("Jonny001/Text-to-Image") + + result = client.predict( + text=text, + selected_model=selected_model, + api_name="/generate_images" + ) + + print(result) + return "Hello, World!" diff --git a/src/routes/router.py b/src/routes/router.py index 9293e2da81b71c3adca2ed4c878dc54359ffbdc9..128917a091b0f64b59166be4ea812d7088a21d5c 100644 --- a/src/routes/router.py +++ b/src/routes/router.py @@ -13,6 +13,7 @@ suggest_controller = SuggestController() def setup_routes(app): @app.route('/glb', methods=['GET']) + # Image to model def get_glb(): base_image_url = request.args.get('base_image_url', type=str) hf_token = request.args.get('hf_token', '', type=str) @@ -29,12 +30,16 @@ def setup_routes(app): slat_sampling_steps, multiimage_algo, mesh_simplify, texture_size) + # Text to image @app.route('/render', methods=['GET']) def get_render(): hf_token = request.args.get('hf_token', '', type=str) + text = request.args.get('text', 'Hello!!', type=str) + selected_model = request.args.get('selected_model', 'Model 1 (Turbo Realism)', type=str) - return render_controller.get_render(hf_token) + return render_controller.get_render(hf_token, text, selected_model) + # Suggest @app.route('/suggest', methods=['GET']) def get_suggest(): hf_token = request.args.get('hf_token', '', type=str)