Spaces:
Runtime error
Runtime error
Commit
•
c63f250
1
Parent(s):
2dcbe60
Update app.py
Browse files
app.py
CHANGED
@@ -194,7 +194,8 @@ class VideoClassifier:
|
|
194 |
except:
|
195 |
prompt1 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
|
196 |
messages = [{"role": "user", "content": prompt1}]
|
197 |
-
|
|
|
198 |
# output = ""
|
199 |
# for response in stream:
|
200 |
# output += response['token'].text
|
@@ -270,7 +271,8 @@ class VideoClassifier:
|
|
270 |
except:
|
271 |
prompt2 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
|
272 |
messages = [{"role": "user", "content": prompt1}]
|
273 |
-
|
|
|
274 |
|
275 |
print("Preprocess Answer",answer)
|
276 |
|
|
|
194 |
except:
|
195 |
prompt1 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
|
196 |
messages = [{"role": "user", "content": prompt1}]
|
197 |
+
stream = self.client.chat_completion(messages, max_tokens=100)
|
198 |
+
main_class = stream.choices[0].message.content.strip()
|
199 |
# output = ""
|
200 |
# for response in stream:
|
201 |
# output += response['token'].text
|
|
|
271 |
except:
|
272 |
prompt2 = template1.format(main_categories=main_categories, transcript=transcript, captions=captions)
|
273 |
messages = [{"role": "user", "content": prompt1}]
|
274 |
+
stream = self.client.chat_completion(messages, max_tokens=100)
|
275 |
+
answer = stream.choices[0].message.content.strip()
|
276 |
|
277 |
print("Preprocess Answer",answer)
|
278 |
|