Update app.py

app.py

@@ -79,7 +79,7 @@ def downsample_video(video_path):
     vidcap.release()
     return frames
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def generate_image(model_name: str, text: str, image: Image.Image,
                    max_new_tokens: int = 1024,
                    temperature: float = 0.6,
@@ -98,7 +98,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
     elif model_name == "GLM-4.1V-9B-Thinking":
         processor = processor_s
         model = model_s
-    elif model_name == "ViLaSR":
+    elif model_name == "ViLaSR-7B":
         processor = processor_y
         model = model_y
     else:
@@ -154,7 +154,7 @@ def generate_video(model_name: str, text: str, video_path: str,
     elif model_name == "GLM-4.1V-9B-Thinking":
         processor = processor_s
         model = model_s
-    elif model_name == "ViLaSR":
+    elif model_name == "ViLaSR-7B":
         processor = processor_y
         model = model_y
     else:
@@ -271,7 +271,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                     markdown_output = gr.Markdown(label="(Result.md)")
 
             model_choice = gr.Radio(
-                choices=["Camel-Doc-OCR-062825", "GLM-4.1V-9B-Thinking", "Megalodon-OCR-Sync-0713", "ViLaSR"],
+                choices=["Camel-Doc-OCR-062825", "GLM-4.1V-9B-Thinking", "Megalodon-OCR-Sync-0713", "ViLaSR-7B"],
                 label="Select Model",
                 value="Camel-Doc-OCR-062825"
             )