rahul7star commited on
Commit
e682860
·
verified ·
1 Parent(s): e68c15a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,8 +1,9 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Initialize the question-answering pipeline
5
- qa_pipeline = pipeline("question-answering")
 
6
 
7
  def answer_question(context, question):
8
  """
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Specify the model explicitly
5
+ model_name = "rahul7star/Rahul-FineTunedLLM-v03"
6
+ qa_pipeline = pipeline("question-answering", model=model_name)
7
 
8
  def answer_question(context, question):
9
  """