Update app.py

app.py

@@ -1,12 +1,21 @@
 import streamlit as st
 import os
-try:
-    import torch
-    import torch.nn as nn
-except ModuleNotFoundError as e:
-    st.error("Dependencies are still being installed. Please wait a moment and refresh the page.")
+
+# Function to safely import a module
+def safe_import(module_name):
+    try:
+        return __import__(module_name)
+    except ImportError:
+        return None
+
+# Try to import torch and other dependencies
+torch = safe_import('torch')
+if torch is None:
+    st.error("Torch is not installed yet. Please wait a moment for the dependencies to install.")
     st.stop()  # Stop the execution of the script
 
+# Import nn from torch
+import torch.nn as nn
 # architecture
 class AddModel(nn.Module):
     def __init__(self):