richardkimsm89 commited on
Commit
0a22b24
·
verified ·
1 Parent(s): 01655b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ pipeline = pipeline(model = "meta-llama/Llama-3.2-1B-Instruct")
5
 
6
  def inference(input):
7
  output = pipeline(input)
8
- return output
9
 
10
  app = gr.Interface(
11
  fn = inference,
 
5
 
6
  def inference(input):
7
  output = pipeline(input)
8
+ return output[0]['generated_text']
9
 
10
  app = gr.Interface(
11
  fn = inference,