Spaces:

stchakman
/

Fridge2Dish

Runtime error

stchakman commited on Mar 24, 2023

Commit

dbd05db

1 Parent(s): 1a79c0c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -56,7 +56,7 @@ def generate_image(prompt):
     pipe = StableDiffusionPipeline.from_pretrained(model_id)
     # If you have a GPU available, uncomment the following line
-    # pipe = pipe.to("cuda")
     image = pipe(prompt).images[0]
     return image

     pipe = StableDiffusionPipeline.from_pretrained(model_id)
     # If you have a GPU available, uncomment the following line
+    pipe = pipe.to("cuda")
     image = pipe(prompt).images[0]
     return image