KishoreK commited on
Commit
46e9920
·
verified ·
1 Parent(s): d06aa40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B-Instruct", tok
8
  def predict(system="you are a helpful assistant", query):
9
  messages = []
10
  messages.append({"role": "system","content": system})
11
- messages.append({"role": "system","content": query})
12
  return pipe(messages,max_new_tokens=256)[0]["generated_text"][-1]
13
 
14
  gradio_app = gr.Interface(
 
8
  def predict(system="you are a helpful assistant", query):
9
  messages = []
10
  messages.append({"role": "system","content": system})
11
+ messages.append({"role": "user","content": query})
12
  return pipe(messages,max_new_tokens=256)[0]["generated_text"][-1]
13
 
14
  gradio_app = gr.Interface(