ML610 commited on
Commit
df8d62b
1 Parent(s): a5a87a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ examples = [example_1, example_2]
70
 
71
  def generate_code(user_input):
72
  response = generate(llm, generation_config, user_input)
73
- return response
74
 
75
  UI = gr.Interface(
76
  fn=generate_code,
 
70
 
71
  def generate_code(user_input):
72
  response = generate(llm, generation_config, user_input)
73
+ return response['text']
74
 
75
  UI = gr.Interface(
76
  fn=generate_code,