JoPmt commited on
Commit
e9503d7
1 Parent(s): 852b160

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -168,8 +168,8 @@ def infer(
168
  output_type="pt",
169
  ).frames
170
 
171
- free_memory()
172
- return (video_pt, seed)
173
 
174
 
175
  def convert_to_gif(video_path):
@@ -196,7 +196,7 @@ def delete_old_files():
196
 
197
 
198
  ##threading.Thread(target=delete_old_files, daemon=True).start()
199
- ##@spaces.GPU
200
  def generate(
201
  prompt,
202
  image_input,
 
168
  output_type="pt",
169
  ).frames
170
 
171
+ ##free_memory()
172
+ return video_pt, seed
173
 
174
 
175
  def convert_to_gif(video_path):
 
196
 
197
 
198
  ##threading.Thread(target=delete_old_files, daemon=True).start()
199
+ @spaces.GPU
200
  def generate(
201
  prompt,
202
  image_input,