Update app.py

app.py

@@ -1,19 +1,22 @@
 import gradio as gr
 
 import os
+
+import requests
+headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+API_URL = "https://api-inference.huggingface.co/models/gpt2"
+def query(payload):
+  response = requests.post(API_URL, headers=headers, json=payload)
+  return response.json()
+        
 def yes_man(message, history):
    # if message.endswith("?"):
     #    return "Yes"
     #else:
     #    return "Ask me anything!"
     
-    import requests
-    headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
-    API_URL = "https://api-inference.huggingface.co/models/gpt2"
-    def query(payload):
-      response = requests.post(API_URL, headers=headers, json=payload)
-      return response.json()
-    data = query({"inputs": message})
+    while not 'generated_text' not in list(data[0].keys()):
+        data = query({"inputs": message})
     return data
 gr.ChatInterface(
     yes_man,