Spaces:

yizhezhu
/

MoMA_demo

Paused

Kunpeng Song commited on May 21

Commit

870ee64

•

1 Parent(s): ef3a17c

random

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ article = "<p style='text-align: center'><a href='https://news.machinelearning.s
 def MoMA_demo(rgb, subject, prompt):
     # move the input and model to GPU for speed if available
     with torch.no_grad():
-        generated_image = model.generate_images(rgb, subject, prompt, strength=1.0, seed=2)
     return generated_image
 def inference(rgb, subject, prompt):

 def MoMA_demo(rgb, subject, prompt):
     # move the input and model to GPU for speed if available
     with torch.no_grad():
+        generated_image = model.generate_images(rgb, subject, prompt, strength=1.0, seed=np.random.randint(0,1000))
     return generated_image
 def inference(rgb, subject, prompt):