t

app.py

@@ -126,10 +126,10 @@ def f(context, temperature, top_p, max_length, model_idx):
             
             # for some reson indexing with 'generated-text' doesn't work
             # edit: maybe because I was using generated-text, not generated_text (note the underscore in the second)
-            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]
+            # try:
+            generated_text = generator(context, max_length=max_length, top_p=top_p, temperature=temperature, num_return_sequences=1)[0][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