Update app.py

app.py

@@ -1,7 +1,12 @@
 import streamlit as st
-import torch
-import torch.nn as nn
 import os
+# Try to import torch and other dependencies
+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.")
+    st.stop()  # Stop the execution of the script
 
 # architecture
 class AddModel(nn.Module):