Namitg02 commited on
Commit
c5ac816
·
verified ·
1 Parent(s): 59d3880

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
- 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)
 
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)