Update main.py

main.py

@@ -10,8 +10,8 @@ client = InferenceClient("FacebookAI/roberta-large-mnli")
 
 class Item(BaseModel):
     prompt: str
-    history: list
-    system_prompt: str
+    #history: list
+    #system_prompt: str
     #temperature: float = 0.0
     #max_new_tokens: int = 1048
     #top_p: float = 0.15
@@ -26,13 +26,6 @@ class Item(BaseModel):
 #    prompt += f"[INST] {message} [/INST]"
 #    return prompt
 
-def format_prompt(message, history):
-    prompt = "<s>"
-    for user_prompt, bot_response in history:
-        prompt += f"[INST] {user_prompt} [/INST]"
-        prompt += f" {bot_response}</s> "
-    prompt += f"[INST] {message} [/INST]"
-    return prompt
 
 def generate(item: Item):
     #temperature = float(item.temperature)
@@ -50,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)
+    #text = format_prompt(f"{item.system_prompt}, {item.prompt}", item.history)
+    text = f"{item.prompt}
+    print(text)
     labels = ["Requirement", "Information"]
     print(labels)
     stream = client.zero_shot_classification(text, labels)