un-index commited on
Commit
4fc4e44
·
1 Parent(s): 1ef28ae
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def f(context, temperature, top_p, max_length, model_idx, SPACE_VERIFICATION_KEY
94
  context = json.loads(response.content.decode("utf-8"))#[0]['generated_text']
95
  # handle inconsistent inference API
96
  if 'generated_text' in context[0]:
97
- context = context[0]['generated_text']
98
  else:
99
  context = context[0][0]['generated_text']
100
  generated_text += context
 
94
  context = json.loads(response.content.decode("utf-8"))#[0]['generated_text']
95
  # handle inconsistent inference API
96
  if 'generated_text' in context[0]:
97
+ context = context[0]['generated_text']
98
  else:
99
  context = context[0][0]['generated_text']
100
  generated_text += context