Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ pipe.to("cpu") # Ensure it runs on CPU
|
|
9 |
|
10 |
def generate_image(prompt, pipe):
|
11 |
# Generate the image using the pipeline
|
12 |
-
image = pipe(prompt
|
13 |
return image
|
14 |
|
15 |
def chatbot(prompt):
|
|
|
9 |
|
10 |
def generate_image(prompt, pipe):
|
11 |
# Generate the image using the pipeline
|
12 |
+
image = pipe(prompt).images[0]
|
13 |
return image
|
14 |
|
15 |
def chatbot(prompt):
|