Update main.py

main.py

@@ -43,8 +43,9 @@ def generate(item: Item):
 
     #formatted_prompt = format_prompt(f"{item.system_prompt}, {item.prompt}", item.history)
     text = item.prompt
+    #print(item.prompt)
     labels = ["Requirement", "Information"]
-    client.zero_shot_classification(text, labels)
+    stream = client.zero_shot_classification(text, labels)
     #stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""