arjunanand13 commited on
Commit
a5c7846
·
verified ·
1 Parent(s): eecc824

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(main_categories=main_categories, transcript=transcript, captions=captions)
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()