Spaces:

Fazzie
/

PokemonGAI

Runtime error

Fazzie commited on Apr 25, 2023

Commit

84da24b

1 Parent(s): 9225c42

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,8 @@ def imagen(inputs):
         pipe = pipe.to("cuda")
     prompt = "a photo of an astronaut riding a horse on mars"
-    image = pipe(prompt=prompt, negative_prompt=negative_prompt).images[0]
     return image

         pipe = pipe.to("cuda")
     prompt = "a photo of an astronaut riding a horse on mars"
+    negative_prompt = 'pixelated, distorted'
+    image = pipe(prompt=inputs, negative_prompt=negative_prompt).images[0]
     return image