bimbambom

app.py

@@ -3,28 +3,27 @@ import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
 
-model_id = "Apocalypse-19/shoe-generator"
+model_id = "CiroN2022/shoes"
 pipe = DiffusionPipeline.from_pretrained(model_id)
 
 pipe.to("cuda")
 
 # Function to generate an image from text using diffusion
 @spaces.GPU
-def generate_image(prompt, neg_prompt):
+def generate_image(prompt):
     
-    images = pipe(prompt, negative_prompt=neg_prompt,)
+    images = pipe(prompt)
     return images[0]
 
 _TITLE = "Shoe Generator"
 with gr.Blocks(_TITLE) as ShoeGen:
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(lines=3, label="Enter a prompt")
-            neg_prompt = gr.Textbox(lines=3, label="Enter a negative prompt")
-        with gr.Column():
+            prompt = gr.Textbox(label="Enter a prompt")
             button_gen = gr.Button("Generate Image")
+        with gr.Column():
             output = gr.Image(label="Generated Image")
 
-    button_gen.click(generate_image, inputs=[prompt, neg_prompt], outputs=output)
+    button_gen.click(generate_image, inputs=[prompt], outputs=output)
 
 ShoeGen.launch()