Update app.py
Browse files
app.py
CHANGED
|
@@ -81,8 +81,25 @@ print("check2")
|
|
| 81 |
print("result")
|
| 82 |
|
| 83 |
print("check3")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 84 |
#chain = pipe(question = question,context = "Use the following information to answer the question. {context}.")
|
| 85 |
-
chain = pipe(question = question,context = "Use the following information to answer the question. Diabetes can be cured by eating apples.")
|
| 86 |
#context = "Use the following information to answer the question. Diabetes can be cured by eating apples."
|
| 87 |
|
| 88 |
print("check3A")
|
|
|
|
| 81 |
print("result")
|
| 82 |
|
| 83 |
print("check3")
|
| 84 |
+
|
| 85 |
+
messages = [
|
| 86 |
+
{
|
| 87 |
+
"role": "system",
|
| 88 |
+
"content": "You are a friendly chatbot who responds in the style of a doctor",
|
| 89 |
+
},
|
| 90 |
+
{"role": "user", "content": "How can I reverse diabetes?"},
|
| 91 |
+
]
|
| 92 |
+
prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
| 93 |
+
chain = pipe(prompt, max_new_tokens=256, do_sample=True)
|
| 94 |
+
print(chain[0]["generated_text"])
|
| 95 |
+
|
| 96 |
+
#chain = pipe(question = question,context = "Use the following information to answer the question. Diabetes can be cured by eating apples.")
|
| 97 |
+
|
| 98 |
+
|
| 99 |
+
|
| 100 |
+
|
| 101 |
+
|
| 102 |
#chain = pipe(question = question,context = "Use the following information to answer the question. {context}.")
|
|
|
|
| 103 |
#context = "Use the following information to answer the question. Diabetes can be cured by eating apples."
|
| 104 |
|
| 105 |
print("check3A")
|