Update app.py

app.py

@@ -17,7 +17,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
 BASETEN_API = os.environ.get("BASETEN_API", None)
 BASETEN_KEY = os.environ.get("BASETEN_KEY", None)
 
-#st.set_page_config(layout="wide")
+st.set_page_config(layout="wide")
 # Load custom CSS to integrate Bootstrap, Font Awesome, and Google Fonts
 st.markdown('''
     <link href="https://fonts.googleapis.com/css?family=Amatic+SC:400,700|Dosis:400,500,700&subset=latin,latin-ext" rel="stylesheet">
@@ -149,7 +149,7 @@ with right:
         }
         response = requests.post(f"{BASETEN_API}", headers=headers, json=payload)
         if response.status_code == 200:
-            st.write(f"**Response:** {response.json().get('response', 'No response')}")
+            st.write(f"**Response:** {response.json()}")
         else:
             st.write("Failed to get a response")