ejschwartz commited on
Commit
051f31b
·
1 Parent(s): 398c3b4
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def predict(code):
53
  )
54
  print(f"Pipe out: {repr(pipe_out)}")
55
 
56
- return pipe_out['generated_text']
57
 
58
  demo = gr.Interface(
59
  fn=predict,
 
53
  )
54
  print(f"Pipe out: {repr(pipe_out)}")
55
 
56
+ return pipe_out[0]['generated_text']
57
 
58
  demo = gr.Interface(
59
  fn=predict,