t

app.py

@@ -129,7 +129,15 @@ def f(context, temperature, top_p, max_length, model_idx, SPACE_VERIFICATION_KEY
             # 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][0]['generated_text']
+            # 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]
+            # [0][0]['generated_text']
+
+            if 'generated_text' in generated_text:
+                return generated_text['generated_text']
+            else:
+                return generated_text[0]['generated_text']
+
             # except: 
                 # generated_text = generator(context, max_length=max_length, top_p=top_p, temperature=temperature, num_return_sequences=1)[0]
                 
@@ -150,7 +158,7 @@ iface = gr.Interface(f, [
     gr.inputs.Slider(
         minimum=20, maximum=512, default=30, label="max length"),
     gr.inputs.Dropdown(["GPT-J-6B", "GPT-2"], type="index", label="model"),
-    gr.inputs.Textbox(lines=1, placeholder="space key", default="xxxxxxxx", label="space verification key")
+    gr.inputs.Textbox(lines=1, placeholder="space key", label="space verification key")
 
 ], outputs="text", title=title, examples=examples)
 iface.launch()  # enable_queue=True