Namitg02 commited on
Commit
0f4fe6a
·
verified ·
1 Parent(s): 0048f5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -74,13 +74,13 @@ qa_chat_prompt = ChatPromptTemplate.from_messages(
74
  ]
75
  )
76
 
77
- model = "HuggingFaceH4/zephyr-7b-beta"
78
- pipe = pipeline(task="text-generation", retriever = retriever,chat = qa_chat_prompt)
79
 
80
 
81
  #chain = qa_chat_prompt | pipe
82
 
83
  import gradio as gr
84
  #ragdemo = gr.load("models/HuggingFaceH4/zephyr-7b-beta")
85
- ragdemo = gr.Interface.from_pipeline(pipe, model)
86
  ragdemo.launch(debug=True)
 
74
  ]
75
  )
76
 
77
+ llm_model = "HuggingFaceH4/zephyr-7b-beta"
78
+ pipe = pipeline(task="text-generation",model = llm_model, retriever = retriever,chat = qa_chat_prompt)
79
 
80
 
81
  #chain = qa_chat_prompt | pipe
82
 
83
  import gradio as gr
84
  #ragdemo = gr.load("models/HuggingFaceH4/zephyr-7b-beta")
85
+ ragdemo = gr.Interface.from_pipeline(pipe)
86
  ragdemo.launch(debug=True)