Update src/model.py

src/model.py

@@ -52,14 +52,15 @@ def describe_video(video, question):
         msgs=msgs,
         tokenizer=tokenizer,
         sampling=True,
-        temperature=0.7,
         stream=True,
+        top_p=0.8,
+        top_k=100,
+        temperature=0.7,
+        repetition_penalty=1.05,
+        max_new_tokens=2048,
         system_prompt="You are an AI assistant specialized in visual content analysis. Given a video and a related question, analyze the video thoroughly and provide a precise and informative answer based on the visible content. Ensure your response is clear, accurate, and directly addresses the question.",
         **params
     )
 
-    # Consume the generator and concatenate the results
-    full_answer = "".join(answer)
-
-    # Return the full answer as a string
-    return full_answer
+    # Return the answer
+    return "".join(answer)