t

app.py

@@ -112,12 +112,15 @@ def f(context, temperature, top_p, max_length, model_idx, SPACE_VERIFICATION_KEY
                     response = requests.request("POST", API_URL, data=json.dumps(payload), headers=headers)
                     context = json.loads(response.content.decode("utf-8"))#[0]['generated_text']
                     # context = get_generated_text(generated_context)
+                    
                     # handle inconsistent inference API
-                    if 'generated_text' in context[0]:
-                        context = context[0]['generated_text']
-                    else:
-                        context = context[0][0]['generated_text']
+                    # if 'generated_text' in context[0]:
+                    #     context = context[0]['generated_text']
+                    # else:
+                    #     context = context[0][0]['generated_text']
                         
+                    context = get_generated_text(context)
+
                     generated_text += context
                     max_length -= 250