Update app.py

app.py

@@ -194,7 +194,8 @@ class VideoClassifier:
             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)
+                stream = self.client.chat_completion(messages, max_tokens=100)
+                main_class = stream.choices[0].message.content.strip()
                 # output = ""
                 # for response in stream:
                 #     output += response['token'].text 
@@ -270,7 +271,8 @@ class VideoClassifier:
             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)
+                stream = self.client.chat_completion(messages, max_tokens=100)
+                answer = stream.choices[0].message.content.strip()
                 
             print("Preprocess Answer",answer)