cocktailpeanut commited on
Commit
d5ff47c
1 Parent(s): 8ef362c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import torch
3
  import os
4
- import spaces
5
  import uuid
6
  import devicetorch
7
 
@@ -43,7 +43,7 @@ from transformers import CLIPFeatureExtractor
43
  feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
44
 
45
  # Function
46
- @spaces.GPU(duration=15,enable_queue=True)
47
  def generate_image(prompt, base, motion, step, progress=gr.Progress()):
48
  global step_loaded
49
  global base_loaded
 
1
  import gradio as gr
2
  import torch
3
  import os
4
+ #import spaces
5
  import uuid
6
  import devicetorch
7
 
 
43
  feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
44
 
45
  # Function
46
+ #@spaces.GPU(duration=15,enable_queue=True)
47
  def generate_image(prompt, base, motion, step, progress=gr.Progress()):
48
  global step_loaded
49
  global base_loaded