kidsampson commited on
Commit
3e733bb
·
verified ·
1 Parent(s): a32272c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,7 +10,9 @@ pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-3B-Instruct")
10
 
11
  def response(messages):
12
  message = ['{"role": "user", "content": '+messages+'},']
13
- return pipe(message)
 
 
14
 
15
  demo = gr.Interface(
16
  response,
 
10
 
11
  def response(messages):
12
  message = ['{"role": "user", "content": '+messages+'},']
13
+ return pipe(message)[1]["conrent"]
14
+
15
+ # [[{'generated_text': '{"role": "user", "content": hi! what is your name?}, {"role": "assistant", "content": "Hi! I\'m an AI assistant, so I'}]]
16
 
17
  demo = gr.Interface(
18
  response,