upd

app.py

@@ -9,7 +9,6 @@ API_TOKEN = st.secrets["hf_api_token"]
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 API_URL = "https://api-inference.huggingface.co/models/arampacha/DialoGPT-medium-simpsons"
 
-
 def query(payload):
     data = json.dumps(payload)
     response = requests.request("POST", API_URL, headers=headers, data=data)
@@ -29,7 +28,7 @@ def fake_query(payload):
 
 parameters = {
     "min_length":None,
-    "max_length":50,
+    "max_length":100,
     "top_p":0.92,
     "repetition_penalty":None,
 }
@@ -54,8 +53,12 @@ def on_input():
         }
         # result = fake_query(payload)
         result = query(payload)
-        st.session_state.update(result["conversation"])
-        st.session_state.full_text += f'_Chatbot_ > {result["generated_text"]}\n\n'
+        try:
+            st.session_state.update(result["conversation"])
+            st.session_state.full_text += f'_Chatbot_ > {result["generated_text"]}\n\n'
+        except:
+            st.write("D'oh! Something went wrong. Try to rerun the app.")
+            st.write(result)
     st.session_state.count += 1
 
 
@@ -100,4 +103,4 @@ dialog_output.markdown(dialog_text)
 def restart():
     st.session_state.clear()
     
-st.button("Restart", on_click=st.session_state.clear)
+st.button("Restart", on_click=st.session_state.clear)