Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import spaces
|
|
| 4 |
import gradio as gr
|
| 5 |
|
| 6 |
# Model and Tokenizer
|
| 7 |
-
model_name = "meta-llama/Llama-
|
| 8 |
|
| 9 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 10 |
|
|
@@ -27,5 +27,5 @@ def generate_text(prompt):
|
|
| 27 |
def chat_with_llama(prompt):
|
| 28 |
return generate_text(prompt)
|
| 29 |
|
| 30 |
-
gr.Interface(fn=chat_with_llama, inputs="text", outputs="text", title="LLaMA
|
| 31 |
|
|
|
|
| 4 |
import gradio as gr
|
| 5 |
|
| 6 |
# Model and Tokenizer
|
| 7 |
+
model_name = "meta-llama/Meta-Llama-3-8B" # Change to 13B or 70B if needed
|
| 8 |
|
| 9 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 10 |
|
|
|
|
| 27 |
def chat_with_llama(prompt):
|
| 28 |
return generate_text(prompt)
|
| 29 |
|
| 30 |
+
gr.Interface(fn=chat_with_llama, inputs="text", outputs="text", title="LLaMA 3 Chatbot").launch()
|
| 31 |
|