Skip to content
Snippets Groups Projects
Commit 308a3e24 authored by BIRK Renaud's avatar BIRK Renaud
Browse files

feat: API GET /render

parent 0d615219
No related branches found
No related tags found
No related merge requests found
import os
from gradio_client import Client, handle_file
class RenderController:
def get_render(self, hf_token='') -> str:
return "Hello, World!"
from flask import Blueprint, request from flask import Blueprint, request
from controllers.glb_controller import GlbController from controllers.glb_controller import GlbController
from controllers.render_controller import RenderController
from controllers.suggest_controller import SuggestController from controllers.suggest_controller import SuggestController
glb_bp = Blueprint('glb', __name__) glb_bp = Blueprint('glb', __name__)
render_bp = Blueprint('render', __name__)
suggest_bp = Blueprint('suggest', __name__) suggest_bp = Blueprint('suggest', __name__)
glb_controller = GlbController() glb_controller = GlbController()
render_controller = RenderController()
suggest_controller = SuggestController() suggest_controller = SuggestController()
def setup_routes(app): def setup_routes(app):
...@@ -26,6 +29,10 @@ def setup_routes(app): ...@@ -26,6 +29,10 @@ def setup_routes(app):
slat_sampling_steps, multiimage_algo, slat_sampling_steps, multiimage_algo,
mesh_simplify, texture_size) mesh_simplify, texture_size)
@app.route('/render', methods=['GET'])
def get_render():
return render_controller.get_render()
@app.route('/suggest', methods=['GET']) @app.route('/suggest', methods=['GET'])
def get_suggest(): def get_suggest():
hf_token = request.args.get('hf_token', '', type=str) hf_token = request.args.get('hf_token', '', type=str)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment