Spaces:
Runtime error
Runtime error
Commit
•
2dcbe60
1
Parent(s):
1b728eb
Update app.py
Browse files
app.py
CHANGED
@@ -192,6 +192,7 @@ class VideoClassifier:
|
|
192 |
chain1 = LLMChain(llm=self.llm, prompt=prompt1)
|
193 |
main_class = chain1.predict(main_categories=main_categories, transcript=transcript, captions=captions)
|
194 |
except:
|
|
|
195 |
messages = [{"role": "user", "content": prompt1}]
|
196 |
main_class = self.client.chat_completion(messages, max_tokens=100)
|
197 |
# output = ""
|
@@ -267,6 +268,7 @@ class VideoClassifier:
|
|
267 |
chain2 = LLMChain(llm=self.llm, prompt=prompt2)
|
268 |
answer = chain2.predict(sub_categories=sub_categories, transcript=transcript, captions=captions,main_class=main_class)
|
269 |
except:
|
|
|
270 |
messages = [{"role": "user", "content": prompt1}]
|
271 |
answer = self.client.chat_completion(messages, max_tokens=100)
|
272 |
|
|
|
192 |
chain1 = LLMChain(llm=self.llm, prompt=prompt1)
|
193 |
main_class = chain1.predict(main_categories=main_categories, transcript=transcript, captions=captions)
|
194 |
except:
|
195 |
+
prompt1 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
|
196 |
messages = [{"role": "user", "content": prompt1}]
|
197 |
main_class = self.client.chat_completion(messages, max_tokens=100)
|
198 |
# output = ""
|
|
|
268 |
chain2 = LLMChain(llm=self.llm, prompt=prompt2)
|
269 |
answer = chain2.predict(sub_categories=sub_categories, transcript=transcript, captions=captions,main_class=main_class)
|
270 |
except:
|
271 |
+
prompt2 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
|
272 |
messages = [{"role": "user", "content": prompt1}]
|
273 |
answer = self.client.chat_completion(messages, max_tokens=100)
|
274 |
|