Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -270,7 +270,7 @@ class VideoClassifier:
|
|
270 |
chain2 = LLMChain(llm=self.llm, prompt=prompt2)
|
271 |
sub_class = chain2.predict(sub_categories=sub_categories, transcript=transcript, captions=captions,main_class=main_class)
|
272 |
except:
|
273 |
-
prompt2 = template2.format(
|
274 |
messages = [{"role": "user", "content": prompt2}]
|
275 |
stream = self.client.chat_completion(messages, max_tokens=100)
|
276 |
sub_class = stream.choices[0].message.content.strip()
|
|
|
270 |
chain2 = LLMChain(llm=self.llm, prompt=prompt2)
|
271 |
sub_class = chain2.predict(sub_categories=sub_categories, transcript=transcript, captions=captions,main_class=main_class)
|
272 |
except:
|
273 |
+
prompt2 = template2.format(sub_categories=sub_categories, transcript=transcript, captions=captions,main_class=main_class)
|
274 |
messages = [{"role": "user", "content": prompt2}]
|
275 |
stream = self.client.chat_completion(messages, max_tokens=100)
|
276 |
sub_class = stream.choices[0].message.content.strip()
|