Geek7 commited on
Commit
4495eec
·
verified ·
1 Parent(s): e70c3c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -54,16 +54,16 @@ async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
54
  # Expose Gradio API
55
  def generate_api(model_str, prompt, seed=1):
56
  result = asyncio.run(infer(model_str, prompt, seed))
57
- result = client.predict(
58
- model_str=model_str,
59
- prompt=prompt,
60
- seed=seed,
61
- api_name="/predict"
62
- )
63
  if result:
64
  return result # Path to generated image
65
  return None
66
 
67
  # Launch Gradio API without frontend
68
- iface = gr.Interface(fn=generate_api, inputs=["text", "text", "number"], outputs="file")
69
  iface.launch(show_api=True, share=True)
 
54
  # Expose Gradio API
55
  def generate_api(model_str, prompt, seed=1):
56
  result = asyncio.run(infer(model_str, prompt, seed))
57
+ # result = client.predict(
58
+ # model_str=model_str,
59
+ # prompt=prompt,
60
+ # seed=seed,
61
+ # api_name="/predict"
62
+ #)
63
  if result:
64
  return result # Path to generated image
65
  return None
66
 
67
  # Launch Gradio API without frontend
68
+ iface = gr.Interface(fn=generate_api, inputs=["text", "text", "number"], outputs="file", api_name="/predict")
69
  iface.launch(show_api=True, share=True)