Torch import fixes and descriptor fixes

Author: adsarver

python/GeminiVideoSummarization/gemini_video_summarization_component/gemini_video_summarization_component.py

@@ -50,9 +50,6 @@
 
 from openai import OpenAI
 
-import torch
-from transformers.models.gemma4.processing_gemma4 import Gemma4Processor
-from transformers.models.gemma4.modeling_gemma4 import Gemma4ForConditionalGeneration
 
 logger = logging.getLogger('GeminiVideoSummarizationComponent')
 
@@ -546,12 +543,9 @@ def _avoid_torchvision_endpoint_frame_index(
 
         num_frames = min(num_frames, available_frame_count)
         while num_frames > 1:
-            indices = torch.arange(
-                0,
-                available_frame_count,
-                available_frame_count / num_frames
-            ).int()
-            if len(indices) <= num_frames and int(indices.max()) < available_frame_count:
+            step = available_frame_count / num_frames
+            indices = [int(index * step) for index in range(num_frames)]
+            if len(indices) <= num_frames and max(indices) < available_frame_count:
                 return num_frames
             num_frames -= 1
 
@@ -664,6 +658,7 @@ def _local_get_response(self, job: mpf.VideoJob, prompt: str) -> str:
 
             inputs = self.processor(**processor_kwargs).to(self.device)
 
+            import torch
             with torch.no_grad():
                 output_ids = self.model.generate(**inputs, max_new_tokens=1024)
 

python/GeminiVideoSummarization/plugin-files/descriptor/descriptor.json

@@ -177,18 +177,18 @@
     },
     "actions": [
         {
-            "name": "GEMINI VIDEO SUMMARIZATION ACTION",
+            "name": "GEMMA VIDEO SUMMARIZATION ACTION",
             "description": "Runs Gemini with a prompt for the provided video.",
             "algorithm": "GeminiVideo",
             "properties": []
         }
     ],
     "tasks": [
         {
-            "name": "GEMINI VIDEO SUMMARIZATION TASK",
+            "name": "GEMMA VIDEO SUMMARIZATION TASK",
             "description": "Runs Gemini video summarization.",
             "actions": [
-                "GEMINI VIDEO SUMMARIZATION ACTION"
+                "GEMMA VIDEO SUMMARIZATION ACTION"
             ]
         }
     ],