yizhezhu commited on
Commit
d376df6
1 Parent(s): 6182092

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ import os
12
  os.environ["CUDA_VISIBLE_DEVICES"]="0"
13
 
14
  title = "MoMA"
15
- description = "This model has to run on GPU. Checkout our project page: https://moma-adapter.github.io/"
16
 
17
  def MoMA_demo(rgb, subject, prompt, strength, seed):
18
  seed = int(seed) if seed else 0
@@ -24,7 +24,7 @@ def MoMA_demo(rgb, subject, prompt, strength, seed):
24
  print(f"Seed: {seed}")
25
 
26
  with torch.no_grad():
27
- generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=seed)
28
  return generated_image
29
 
30
  def inference(rgb, subject, prompt, strength, seed):
 
12
  os.environ["CUDA_VISIBLE_DEVICES"]="0"
13
 
14
  title = "MoMA"
15
+ description = "This space has been migrated to Zero_GPU: https://huggingface.co/spaces/yizhezhu/MoMA_zeroGPU"
16
 
17
  def MoMA_demo(rgb, subject, prompt, strength, seed):
18
  seed = int(seed) if seed else 0
 
24
  print(f"Seed: {seed}")
25
 
26
  with torch.no_grad():
27
+ generated_image = model.generate_images()
28
  return generated_image
29
 
30
  def inference(rgb, subject, prompt, strength, seed):