Update app.py

app.py

@@ -1,12 +1,12 @@
 import torch
-torch.jit.script = lambda f: f
+# torch.jit.script = lambda f: f
 # General
 import os
 from os.path import join as opj
 import argparse
 import datetime
 from pathlib import Path
-import spaces
+# import spaces
 import gradio as gr
 import tempfile
 import yaml
@@ -56,7 +56,7 @@ msxl_model = init_v2v_model(cfg_v2v, device)
 # -------------------------
 # ----- Functionality -----
 # -------------------------
-@spaces.GPU(duration=120)
+# @spaces.GPU(duration=120)
 def generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, seed, t, image_guidance, where_to_log=result_fol):
     now = datetime.datetime.now()
     name = prompt[:100].replace(" ", "_") + "_" + str(now.time()).replace(":", "_").replace(".", "_")
@@ -86,7 +86,7 @@ def generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, se
     video_path = opj(where_to_log, name+".mp4")
     return video_path
 
-@spaces.GPU(duration=400)
+# @spaces.GPU(duration=400)
 def enhance(prompt, input_to_enhance, num_frames=None, image=None, model_name_stage1=None, model_name_stage2=None, seed=33, t=50, image_guidance=9.5, result_fol=result_fol):
     if input_to_enhance is None:
         input_to_enhance = generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, seed, t, image_guidance)