Update app.py

app.py

@@ -268,22 +268,22 @@ class VideoClassifier:
         if self.model_name=='mistral':
             try:
                 chain2 = LLMChain(llm=self.llm, prompt=prompt2)
-                answer = chain2.predict(sub_categories=sub_categories, transcript=transcript, captions=captions,main_class=main_class)
+                sub_class = chain2.predict(sub_categories=sub_categories, transcript=transcript, captions=captions,main_class=main_class)
             except:
-                prompt2 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
-                messages = [{"role": "user", "content": prompt1}]
+                prompt2 = template2.format(main_categories=main_categories, transcript=transcript, captions=captions)
+                messages = [{"role": "user", "content": prompt2}]
                 stream = self.client.chat_completion(messages, max_tokens=100)
-                answer = stream.choices[0].message.content.strip()
+                sub_class = stream.choices[0].message.content.strip()
                 
-            print("Preprocess Answer",answer)
+            print("Preprocess Answer",sub_class)
 
             try:
                 pattern = r"Sub-class =>\s*(.+)"
-                match = re.search(pattern, answer)
+                match = re.search(pattern, sub_class)
                 if match:
                     sub_class = match.group(1).strip()
             except:
-                subclass=answer
+                subclass=sub_class
         else:
             prompt_text2 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
             response = self.genai_model.generate_content(contents=prompt_text2)