Update app.py
Browse files
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()
|