t

app.py

@@ -131,14 +131,17 @@ def f(context, temperature, top_p, max_length, model_idx, SPACE_VERIFICATION_KEY
             # edit: maybe because I was using generated-text, not generated_text (note the underscore in the second)
             # try:
             # NOTE sometimes it seems to contain another array, weird
-            generated_text = generator(context, max_length=max_length, top_p=top_p, temperature=temperature, num_return_sequences=1)[0]
+            try:
+                generated_text = generator(context, max_length=max_length, top_p=top_p, temperature=temperature, num_return_sequences=1)
+            except Exception as e:
+                return "Exception while generating text: " + str(e)
             # [0][0]['generated_text']
 
             try:
-                if 'generated_text' in generated_text:
-                    return generated_text['generated_text']
-                else:
+                if 'generated_text' in generated_text[0]:
                     return generated_text[0]['generated_text']
+                else:
+                    return generated_text[0][0]['generated_text']
             except:
                 return generated_text