Spaces:

thelip
/

demm

Runtime error

thelip commited on Nov 10, 2024

Commit

fe52cbc

verified ·

1 Parent(s): d91393a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 # Load the LLaMA-2 model and tokenizer from Hugging Face
 model_name = "meta-llama/Llama-2-7b-hf"  # Change to the desired LLaMA model
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 # Load the LLaMA-2 model and tokenizer from Hugging Face
 model_name = "meta-llama/Llama-2-7b-hf"  # Change to the desired LLaMA model
 tokenizer = AutoTokenizer.from_pretrained(model_name)