Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from huggingface_hub import login
|
|
| 9 |
login(token=os.environ["HUGGINGFACE_TOKEN"])
|
| 10 |
|
| 11 |
# Model and Tokenizer
|
| 12 |
-
model_name = "meta-llama/
|
| 13 |
|
| 14 |
tokenizer = AutoTokenizer.from_pretrained(model_name,use_auth_token=True)
|
| 15 |
|
|
@@ -37,5 +37,5 @@ def generate_text(prompt):
|
|
| 37 |
def chat_with_llama(prompt):
|
| 38 |
return generate_text(prompt)
|
| 39 |
|
| 40 |
-
gr.Interface(fn=chat_with_llama, inputs="text", outputs="text", title="LLaMA
|
| 41 |
|
|
|
|
| 9 |
login(token=os.environ["HUGGINGFACE_TOKEN"])
|
| 10 |
|
| 11 |
# Model and Tokenizer
|
| 12 |
+
model_name = "meta-llama/Llama-2-7b" # Change to 13B or 70B if needed
|
| 13 |
|
| 14 |
tokenizer = AutoTokenizer.from_pretrained(model_name,use_auth_token=True)
|
| 15 |
|
|
|
|
| 37 |
def chat_with_llama(prompt):
|
| 38 |
return generate_text(prompt)
|
| 39 |
|
| 40 |
+
gr.Interface(fn=chat_with_llama, inputs="text", outputs="text", title="LLaMA 2 Chatbot").launch()
|
| 41 |
|