t

app.py

@@ -123,7 +123,13 @@ def f(context, temperature, top_p, max_length, model_idx):
             #
             set_seed(randint(1, 2**31))
             # return sequences specifies how many to return
-            return generator(context, max_length=max_length, top_p=top_p, temperature=temperature, num_return_sequences=1)[0]#['generated-text']
+            
+            try:
+                generated_text = generator(context, max_length=max_length, top_p=top_p, temperature=temperature, num_return_sequences=1)[0]['generated_text']
+            except: 
+                generated_text = generator(context, max_length=max_length, top_p=top_p, temperature=temperature, num_return_sequences=1)[0]
+                
+            return generated_text
             # args found in the source: https://github.com/huggingface/transformers/blob/27b3031de2fb8195dec9bc2093e3e70bdb1c4bff/src/transformers/generation_tf_utils.py#L348-L376
 
             # use fallback gpt-2 inference api for this as well