Update app.py

app.py

@@ -69,15 +69,12 @@ def predict_beta(message, chatbot=[], temperature=0.9, max_new_tokens=256, top_p
             raise gr.Error(json_obj['error'] + ' Please refresh and try again with smaller input prompt')
         else:
             warning_msg = f"Unexpected response: {json_obj}"
-            print(warning_msg)
-            raise ValueError(warning_msg)
-    except requests.HTTPError:
-        error_msg = f"Request failed with status code {response.status_code}"
-        print(error_msg)
+            raise gr.Error(warning_msg)
+    except requests.HTTPError as e:
+        error_msg = f"Request failed with status code {e.response.status_code}"
         raise gr.Error(error_msg)
-    except json.JSONDecodeError:
-        error_msg = f"Failed to decode response as JSON: {response.text}"
-        print(error_msg)
+    except json.JSONDecodeError as e:
+        error_msg = f"Failed to decode response as JSON: {str(e)}"
         raise gr.Error(error_msg)
 
 
@@ -110,15 +107,12 @@ def predict_alpha(message, chatbot=[], temperature=0.9, max_new_tokens=256, top_
             raise gr.Error(json_obj['error'] + ' Please refresh and try again with smaller input prompt')
         else:
             warning_msg = f"Unexpected response: {json_obj}"
-            print(warning_msg)
-            raise ValueError(warning_msg)
-    except requests.HTTPError:
-        error_msg = f"Request failed with status code {response.status_code}"
-        print(error_msg)
+            raise gr.Error(warning_msg)
+    except requests.HTTPError as e:
+        error_msg = f"Request failed with status code {e.response.status_code}"
         raise gr.Error(error_msg)
-    except json.JSONDecodeError:
-        error_msg = f"Failed to decode response as JSON: {response.text}"
-        print(error_msg)
+    except json.JSONDecodeError as e:
+        error_msg = f"Failed to decode response as JSON: {str(e)}"
         raise gr.Error(error_msg)
 
 
@@ -162,6 +156,17 @@ Welcome to the Zephyr Playground! This interactive space lets you experience the
 Note: All rights, including licensing and acceptable use policies, related to the Zephyr models, can be found on their respective model pages on Hugging Face.
 """
 
+
+css = """
+.gradio-container {
+    width: 100vw !important;
+    min-height: 100vh !important;
+    padding:0 !important;
+    margin:0 !important;
+    max-width: none !important;
+}
+"""
+
 # Create chatbot components
 chat_beta = gr.Chatbot(label="zephyr-7b-beta")
 chat_alpha = gr.Chatbot(label="zephyr-7b-alpha")
@@ -174,7 +179,7 @@ retry = gr.Button('🔄Retry', variant='secondary')
 undo = gr.Button('↩️Undo', variant='secondary')
 
 # Layout the components using Gradio Blocks API
-with gr.Blocks() as demo:
+with gr.Blocks(css=css) as demo:
   gr.HTML(f'<h1><center> {title} </center></h1>')
   gr.Markdown(description)
   with gr.Row():