Update README.md

README.md

@@ -21,9 +21,13 @@ pipeline = transformers.pipeline(
     trust_remote_code=True,
     device_map="auto",
 )
+
+
+input_prompt = "Question: Are group 2 innate lymphoid cells ( ILC2s ) increased in chronic rhinosinusitis with nasal polyps or eosinophilia?\nAnswer:"
+
 sequences = pipeline(
-   "Question: Are group 2 innate lymphoid cells ( ILC2s ) increased in chronic rhinosinusitis with nasal polyps or eosinophilia?\nAnswer:",
-    max_length=1000,
+    input_prompt,
+    max_length=300,
     do_sample=True,
     top_k=10,
     num_return_sequences=1,
@@ -31,7 +35,8 @@ sequences = pipeline(
 )
 
 for seq in sequences:
-    generated_text = "Question: " + " ".join(seq['generated_text'].split("Question:")[:2])
-    print(f"Result: {generated_text}")
+    generated_text = seq['generated_text'].split("\nQuestion:")[0]
+    generated_text = generated_text.replace(input_prompt, "").strip()
+    print(generated_text)
 
 ```