Update app.py

app.py

@@ -1,22 +1,9 @@
 import gradio as gr
-import requests
+from generate import generate_image
 
-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
+def generate_image_wrapper(prompt, seed=0, negative_prompt="", model="Deliberate", sampler="k_dpmpp_2s_a", steps=50):
+    url = generate_image(prompt, seed=seed, negative_prompt=negative_prompt, model=model, sampler=sampler, steps=steps)
+    return {"image": url}
 
 inputs = [
     gr.inputs.Textbox(label="Prompt"),
@@ -27,9 +14,9 @@ inputs = [
     gr.inputs.Number(label="Steps", default=50)
 ]
 
-outputs = gr.outputs.Image(label="Generated Image")
+outputs = gr.outputs.Image(type="pil")
 
-interface = gr.Interface(generate_image, inputs, outputs, title="Generate Images from Prompts", 
+interface = gr.Interface(generate_image_wrapper, 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"],