Update main.py

main.py

@@ -53,7 +53,9 @@ def generate(item: Item):
     text = item.prompt
     print(item.prompt)
     labels = ["Requirement", "Information"]
+    print(labels)
     stream = client.zero_shot_classification(text, labels)
+    print(stream)
     #stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""