Namitg02 commited on
Commit
10d72c0
·
verified ·
1 Parent(s): 5f04ae4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,14 +77,14 @@ qa_chat_prompt = ChatPromptTemplate.from_messages(
77
 
78
  llm_model = "HuggingFaceH4/zephyr-7b-beta"
79
  print("check2")
80
- pipe = pipeline(task="text-generation",model = llm_model, retriever = retriever,chat = qa_chat_prompt)
81
  print("check3")
82
 
83
  #chain = qa_chat_prompt | pipe
84
 
85
  import gradio as gr
86
  #ragdemo = gr.load("models/HuggingFaceH4/zephyr-7b-beta")
87
- ragdemo = gr.Interface.from_pipeline(pipe)
88
  print("check4")
89
  ragdemo.launch(debug=True)
90
  print("check5")
 
77
 
78
  llm_model = "HuggingFaceH4/zephyr-7b-beta"
79
  print("check2")
80
+ pipe = pipeline(task="text-generation", retriever = retriever,chat = qa_chat_prompt)
81
  print("check3")
82
 
83
  #chain = qa_chat_prompt | pipe
84
 
85
  import gradio as gr
86
  #ragdemo = gr.load("models/HuggingFaceH4/zephyr-7b-beta")
87
+ ragdemo = gr.Interface.from_pipeline(pipe,llm_model)
88
  print("check4")
89
  ragdemo.launch(debug=True)
90
  print("check5")