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

feat: API GET /render to use adapter model

parent d22f6b12
No related branches found
No related tags found
No related merge requests found
import os
from gradio_client import Client
from gradio_client import Client, handle_file
class RenderController:
def get_render(self, hf_token='', text="Hello!!",
selected_model="Model 1 (Turbo Realism)") -> str:
def get_render(self, hf_token='', base_image_url='',
prompt='A realistic photo of a building',
negative_prompt="sketch, drawing",
instruct_query='use everything from the image',
scale=0.8,
nfaa_negative_prompt='naked, bikini, skimpy, scanty, bare skin, lingerie, swimsuit, exposed, see-through') -> str:
client = Client("Jonny001/Text-to-Image")
client = Client("unity/IP-Adapter-Instruct")
result = client.predict(
text=text,
selected_model=selected_model,
api_name="/generate_images"
images=handle_file(base_image_url),
prompt=prompt,
negative_prompt=negative_prompt,
instruct_query=instruct_query,
scale=scale,
nfaa_negative_prompt=nfaa_negative_prompt,
api_name="/generate_image"
)
print(result)
return str(result)
return result
......@@ -11,12 +11,12 @@ glb_controller = GlbController()
render_controller = RenderController()
suggest_controller = SuggestController()
def setup_routes(app):
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)
base_image_url = request.args.get('base_image_url', type=str)
ss_guidance_strength = request.args.get('ss_guidance_strength', 7.5, type=float)
ss_sampling_steps = request.args.get('ss_sampling_steps', 12, type=int)
slat_guidance_strength = request.args.get('slat_guidance_strength', 3, type=float)
......@@ -34,10 +34,15 @@ def setup_routes(app):
@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, text, selected_model)
base_image_url = request.args.get('base_image_url', '', type=str)
prompt = request.args.get('prompt', 'A realistic photo of a building', type=str)
negative_prompt = request.args.get('negative_prompt', 'sketch, drawing', type=str)
instruct_query = request.args.get('instruct_query', 'use everything from the image', type=str)
scale = request.args.get('scale', 0.8, type=float)
nfaa_negative_prompt = request.args.get('nfaa_negative_prompt', 'naked, bikini, skimpy, scanty, bare skin, lingerie, swimsuit, exposed, see-through', type=str)
return render_controller.get_render(hf_token, base_image_url, prompt, negative_prompt,
instruct_query, scale, nfaa_negative_prompt)
# Suggest
@app.route('/suggest', methods=['GET'])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment