Update app.py

app.py

@@ -192,6 +192,7 @@ class VideoClassifier:
                 chain1 = LLMChain(llm=self.llm, prompt=prompt1)
                 main_class  = chain1.predict(main_categories=main_categories, transcript=transcript, captions=captions)
             except:
+                prompt1 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
                 messages = [{"role": "user", "content": prompt1}]
                 main_class = self.client.chat_completion(messages, max_tokens=100)
                 # output = ""
@@ -267,6 +268,7 @@ class VideoClassifier:
                 chain2 = LLMChain(llm=self.llm, prompt=prompt2)
                 answer = 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}]
                 answer = self.client.chat_completion(messages, max_tokens=100)