Update app.py
Browse files
app.py
CHANGED
@@ -78,13 +78,14 @@ qa_chat_prompt = ChatPromptTemplate.from_messages(
|
|
78 |
llm_model = "deepset/roberta-base-squad2"
|
79 |
from transformers import AutoTokenizer
|
80 |
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
81 |
-
question = "How can I reverse diabetes?"
|
|
|
82 |
print("check2")
|
83 |
pipe = pipeline(model = llm_model, tokenizer = tokenizer, task = "question-answering", temperature=0.2)
|
84 |
#"question-answering", "conversational"
|
85 |
|
86 |
print("check3")
|
87 |
-
chain = pipe(question =
|
88 |
#(question = question, context = context)
|
89 |
print("check3A")
|
90 |
import gradio as gr
|
|
|
78 |
llm_model = "deepset/roberta-base-squad2"
|
79 |
from transformers import AutoTokenizer
|
80 |
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
81 |
+
#question = "How can I reverse diabetes?"
|
82 |
+
|
83 |
print("check2")
|
84 |
pipe = pipeline(model = llm_model, tokenizer = tokenizer, task = "question-answering", temperature=0.2)
|
85 |
#"question-answering", "conversational"
|
86 |
|
87 |
print("check3")
|
88 |
+
chain = pipe(question = "How can I reverse diabetes?",context = "You should reply like a doctor")
|
89 |
#(question = question, context = context)
|
90 |
print("check3A")
|
91 |
import gradio as gr
|