Update app.py

app.py

@@ -186,9 +186,17 @@ def chat_with_model(prompt, document_section, model_choice='Llama-2-7b-chat-hf')
             if len(r.token.text) > 0:
                 result="".join(report).strip()
                 res_box.markdown(f'*{result}*')
-                
-        except:
-            st.write('.')
+        except HfHubHTTPError as e:
+            if e.response.status_code == 502:
+                print("Server Error: Bad Gateway. Please try again later.")
+            else:
+                # Handle other HTTP errors
+                print(f"An HTTP error occurred: {e}")
+        except Exception as e:
+            # Handle any other exceptions
+            print(f"An error occurred: {e}")
+
+
 
     full_reply_content = result
     st.write("Elapsed time:")