Spaces:

Bunpheng
/

test-llama2

Runtime error

Bunpheng commited on Sep 18, 2024

Commit

a4d2712

verified ·

1 Parent(s): caec972

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,5 @@
 import gradio as gr
 # gr.load("models/meta-llama/Llama-2-7b").launch()
 self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
 self.model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")

 import gradio as gr
+from transformers import AutoModel, AutoTokenizer
 # gr.load("models/meta-llama/Llama-2-7b").launch()
 self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
 self.model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")