eduard76 commited on
Commit
04757ef
·
verified ·
1 Parent(s): 70332fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-new", 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:"])
 
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:"])