fix

process.py

@@ -1,4 +1,5 @@
 import io
+import spaces
 import argparse
 import numpy as np
 import torch
@@ -67,6 +68,7 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 ).eval()
 
+@spaces.GPU
 def predict(prompt, video_data, temperature):
     strategy = 'chat'
     video = load_video(video_data, strategy=strategy)

requirements.txt

@@ -22,4 +22,5 @@ gevent
 requests
 gradio
 accelerate
-bitsandbytes>=0.39.0
+bitsandbytes>=0.39.0
+spaces