zerostratos commited on
Commit
77d76dd
·
verified ·
1 Parent(s): e059db3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,9 +1,10 @@
1
  import gradio as gr
2
- #from llama_cpp import Llama
3
- from transformers import AutoModelForCausalLM
4
  # Load the model
5
- llm = AutoModelForCausalLM.from_pretrained("Viet-Mistral/Vistral-7B-Chat",trust_remote_code=True)
6
- # filename="models-7B-F16.gguf"
 
7
 
8
  # Define the function to interact with the model
9
  def chat_with_model(user_input):
 
1
  import gradio as gr
2
+ from llama_cpp import Llama
3
+ #from transformers import AutoModelForCausalLM
4
  # Load the model
5
+ llm = Llama.from_pretrained(
6
+ repo_id = "uonlp/Vistral-7B-Chat-gguf",
7
+ filename="ggml-vistral-7B-chat-f16.gguf"
8
 
9
  # Define the function to interact with the model
10
  def chat_with_model(user_input):