Namitg02 commited on
Commit
23f23be
·
verified ·
1 Parent(s): 3318ce2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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 = question,context = retriever)
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