t

app.py

@@ -100,7 +100,7 @@ def f(context, temperature, top_p, max_length, model_idx, SPACE_VERIFICATION_KEY
 
         # maybe try "0" instead or 1, or "1"
         # use GPT-J-6B
-        if model_idx == 1:
+        if model_idx == 0:
             if main_gpt_j_api_up:
                 # for this api, a length of > 250 instantly errors, so use a while loop or something
                 # that would fetch results in chunks of 250
@@ -154,7 +154,7 @@ def f(context, temperature, top_p, max_length, model_idx, SPACE_VERIFICATION_KEY
             response = requests.post(
                 "http://api.vicgalle.net:5000/generate", params=payload).json()
             return response['text']
-        elif model_idx == 0:
+        elif model_idx == 1:
             # use GPT-2
             #
             try:
@@ -246,7 +246,7 @@ iface = gr.Interface(f, [
     top_p,
     gr.inputs.Slider(
         minimum=20, maximum=512, default=30, label="max length"),
-    gr.inputs.Dropdown([ "GPT2","GPT-J-6B", "DistilGPT2", "GPT-Large", "GPT-Neo-2.7B"], type="index", label="model"),
+    gr.inputs.Dropdown(["GPT-J-6B", "GPT2", "DistilGPT2", "GPT-Large", "GPT-Neo-2.7B"], type="index", label="model", default="GPT2"),
     gr.inputs.Textbox(lines=1, placeholder="xxxxxxxx", label="space verification key")
 
 ], outputs="text", title=title, examples=examples, enable_queue = True) # deprecated iwthin iface.launch: https://discuss.huggingface.co/t/is-there-a-timeout-max-runtime-for-spaces/12979/3?u=un-index