Update main.py

main.py

@@ -48,12 +48,16 @@ def generate(item: Item):
     print(text)
     labels = ["Requirement", "Information"]
     print(labels)
-    stream = client.zero_shot_classification(text, labels)
-    print("Stream: " + stream)
+    result = client.zero_shot_classification("The car shall be slow.", labels)
+    print("Predicted Labels:")
+    print(result["labels"][0], result["scores"][0])
+    print(result["labels"][1], result["scores"][1])
+    #stream = client.zero_shot_classification(text, labels)
+    #print("Stream: " + stream)
     #stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
-    for response in stream:
+    for response in result:
         output += response.token.text
     return output