lcbwhite commited on
Commit
2af4db3
·
verified ·
1 Parent(s): 1d48616

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoModelForCausalLM
5
  def LMmodel(message, history):
6
  base_model_id="LeoLM/leo-mistral-hessianai-7b-chat"
7
  base_model = "AutoModelForCausalLM.from_pretrained(base_model_id, device_map="auto",trust_remote_code=True)
8
- return works
9
 
10
 
11
  gr.ChatInterface(LMmodel).launch()
 
5
  def LMmodel(message, history):
6
  base_model_id="LeoLM/leo-mistral-hessianai-7b-chat"
7
  base_model = "AutoModelForCausalLM.from_pretrained(base_model_id, device_map="auto",trust_remote_code=True)
8
+ return "works"
9
 
10
 
11
  gr.ChatInterface(LMmodel).launch()