Update app.py

app.py

@@ -88,7 +88,7 @@ def reset_components():
     return ['pla', 'gnd', 'ver', 'ans'], 0, 256
 
 
-@spaces.GPU
+@spaces.GPU(duration=300)
 def main(video, prompt, role, temperature, max_new_tokens, model, processor):
     history = []
 
@@ -161,6 +161,8 @@ def main(video, prompt, role, temperature, max_new_tokens, model, processor):
         data = processor(text=[text], images=images, videos=videos, return_tensors='pt')
         data = data.to(next(model.parameters()).device)
 
+        print(next(model.parameters()).device)
+
         model.base_model.disable_adapter_layers()
         model.base_model.enable_adapter_layers()
         model.set_adapter('planner')