Spaces:

MUTSC
/

Text-to-Image

Runtime error

MUTSC commited on Apr 16, 2024

Commit

a85d628

verified ·

1 Parent(s): 56914ca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ def generate_image(prompt):
     with torch.no_grad():
         image = model(prompt).images[0]
     # Convert to PIL Image to display in Gradio
-    image = Image.fromarray(image.numpy())
     return image
 # Create a Gradio interface

     with torch.no_grad():
         image = model(prompt).images[0]
     # Convert to PIL Image to display in Gradio
+    #image = Image.fromarray(image.numpy())
     return image
 # Create a Gradio interface