Update app.py

app.py

@@ -1,9 +1,15 @@
 import gradio as gr
 import torch
+
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
+
 import numpy as np
 from transformers import AutoProcessor, AutoModel
 from PIL import Image
-from decord import VideoReader, cpu
+from decord import VideoReader, gpu
 
 def sample_uniform_frame_indices(clip_len, seg_len):
     """
@@ -21,7 +27,7 @@ def sample_uniform_frame_indices(clip_len, seg_len):
     return np.array(indices).astype(np.int64)
 
 def read_video_decord(file_path, indices):
-    vr = VideoReader(file_path, num_threads=1, ctx=cpu(0))
+    vr = VideoReader(file_path, num_threads=1, ctx=gpu(0))
     video = vr.get_batch(indices).asnumpy()
     return video
 
@@ -62,6 +68,7 @@ def model_interface(uploaded_video, model_choice, activities):
 
     processor = AutoProcessor.from_pretrained(model_choice)
     model = AutoModel.from_pretrained(model_choice)
+    model = model.to("cuda")
     
     activities_list = activities.split(",")
     inputs = processor(