Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ from transformers import pipeline
|
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
# Load your HF model (must be on Hub)
|
| 5 |
-
pipe = pipeline("text-generation", model="eduard76/Llama3-8b-good
|
| 6 |
|
| 7 |
def chat_fn(message, history):
|
| 8 |
full_input = "\n".join([f"User: {turn[0]}\nAI: {turn[1]}" for turn in history] + [f"User: {message}\nAI:"])
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
# Load your HF model (must be on Hub)
|
| 5 |
+
pipe = pipeline("text-generation", model="eduard76/Llama3-8b-good", trust_remote_code=True)
|
| 6 |
|
| 7 |
def chat_fn(message, history):
|
| 8 |
full_input = "\n".join([f"User: {turn[0]}\nAI: {turn[1]}" for turn in history] + [f"User: {message}\nAI:"])
|