Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -44,12 +44,12 @@ def generate(item: Item):
|
|
44 |
|
45 |
#formatted_prompt = format_prompt(f"{item.system_prompt}, {item.prompt}", item.history)
|
46 |
#text = format_prompt(f"{item.system_prompt}, {item.prompt}", item.history)
|
47 |
-
text =
|
48 |
print(text)
|
49 |
labels = ["Requirement", "Information"]
|
50 |
print(labels)
|
51 |
stream = client.zero_shot_classification(text, labels)
|
52 |
-
print(stream)
|
53 |
#stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
54 |
output = ""
|
55 |
|
|
|
44 |
|
45 |
#formatted_prompt = format_prompt(f"{item.system_prompt}, {item.prompt}", item.history)
|
46 |
#text = format_prompt(f"{item.system_prompt}, {item.prompt}", item.history)
|
47 |
+
text = item.prompt
|
48 |
print(text)
|
49 |
labels = ["Requirement", "Information"]
|
50 |
print(labels)
|
51 |
stream = client.zero_shot_classification(text, labels)
|
52 |
+
print("Stream: " + stream)
|
53 |
#stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
54 |
output = ""
|
55 |
|