Update app.py

app.py

@@ -206,9 +206,11 @@ class VideoClassifier:
         # print(f"Current model in use: {self.model}")
         if self.model_name=='mistral':
             try:
+                print("Entering mistral chain approach")
                 chain1 = LLMChain(llm=self.llm, prompt=prompt1)
                 main_class  = chain1.predict(main_categories=main_categories, transcript=transcript, captions=captions)
             except:
+                print("Entering mistral template approach")
                 prompt1 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
                 messages = [{"role": "user", "content": prompt1}]
                 stream = self.client.chat_completion(messages, max_tokens=100)