Update main.py

main.py

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