swapnasg commited on
Commit
53a3eee
·
verified ·
1 Parent(s): fe5503a

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
4
- # Load Mistral 7B instruct model
5
  model_id = "mistralai/Mistral-7B-Instruct-v0.1"
6
 
7
  tokenizer = AutoTokenizer.from_pretrained(model_id)
@@ -17,5 +16,4 @@ def chat_with_expert(message, history):
17
  return history, history
18
 
19
  chatbot = gr.ChatInterface(fn=chat_with_expert, title="Expert Chat Assistant")
20
-
21
  chatbot.launch()
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
 
4
  model_id = "mistralai/Mistral-7B-Instruct-v0.1"
5
 
6
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
16
  return history, history
17
 
18
  chatbot = gr.ChatInterface(fn=chat_with_expert, title="Expert Chat Assistant")
 
19
  chatbot.launch()