Update app.py

app.py

@@ -1,9 +1,22 @@
 import gradio as gr
-from generate import generate
+import requests
 
-def generate_wrapper(prompt, seed=0, negative_prompt="", model="Deliberate", sampler="k_dpmpp_2s_a", steps=50):
-    url = generate(prompt, seed=seed, negative_prompt=negative_prompt, model=model, sampler=sampler, steps=steps)
-    return {"image": url}
+api_url = "https://5cb20b40-572c-426f-9466-995256f9b6eb.id.repl.co/generate_image"
+
+def generate_image(prompt, seed=0, negative_prompt="", model="Deliberate", sampler="k_dpmpp_2s_a", steps=50):
+    data = {
+        "prompt": prompt,
+        "seed": seed,
+        "negative_prompt": negative_prompt,
+        "model": model,
+        "sampler": sampler,
+        "steps": steps
+    }
+    response = requests.post(api_url, data=data)
+    if response.status_code == 200:
+        return response.json()["url"]
+    else:
+        return None
 
 inputs = [
     gr.inputs.Textbox(label="Prompt"),
@@ -14,9 +27,9 @@ inputs = [
     gr.inputs.Number(label="Steps", default=50)
 ]
 
-outputs = gr.outputs.Image(type="pil")
+outputs = gr.outputs.Image(label="Generated Image", type="pil")
 
-interface = gr.Interface(generate_wrapper, inputs, outputs, title="Generate Images from Prompts", 
+interface = gr.Interface(generate_image, inputs, outputs, title="Generate Images from Prompts", 
                          description="Enter a prompt and generate an image using this API.", 
                          examples=[["A cat sitting on a chair"], 
                                    ["A landscape with mountains and a lake"],