Update src/text_processor.py

src/text_processor.py

@@ -4,6 +4,7 @@ from pydantic import BaseModel
 import json
 import warnings
 import spaces
+from typing import Dict
 
 # Ignore warnings
 warnings.filterwarnings(action='ignore')
@@ -34,6 +35,7 @@ generation_args = {
     "do_sample": True
 }
 
+@spaces.GPU(duration=75)
 class LLMHelper:
     def __init__(self, pipeline):
         self.chatbot = pipeline
@@ -80,7 +82,7 @@ class VideoAnalysis(BaseModel):
 # Create an instance of LLMHelper (using the already loaded pipeline)
 llm_helper = LLMHelper(pipe) 
 
-def process_llm_output(description -> Dict:
+def process_llm_output(description) -> Dict:
     # Generate the logic from the LLM output
     generated_logic = llm_helper.generate_logic(description