Geek7 commited on
Commit
7eec8a0
1 Parent(s): eb0bba6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -55,6 +55,18 @@ def generate_api(model_str, prompt, seed=1):
55
  return result # Path to generated image
56
  return None
57
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  # Launch Gradio API without frontend
59
  iface = gr.Interface(fn=generate_api, inputs=["text", "text", "number"], outputs="file")
60
  iface.launch(show_api=True, share=True)
 
55
  return result # Path to generated image
56
  return None
57
 
58
+ from gradio_client import Client
59
+
60
+ client = Client("Geek7/mdztxi2")
61
+ result = client.predict(
62
+ model_str=model_str,
63
+ prompt=prompt,
64
+ seed=seed,
65
+ api_name="/generate_api "
66
+ )
67
+
68
+
69
+
70
  # Launch Gradio API without frontend
71
  iface = gr.Interface(fn=generate_api, inputs=["text", "text", "number"], outputs="file")
72
  iface.launch(show_api=True, share=True)