Update app.py

app.py

@@ -1,21 +1,19 @@
 import streamlit as st
 import os
 
-# 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
+    st.stop()
 
-# Import nn from torch
 import torch.nn as nn
+
 # architecture
 class AddModel(nn.Module):
     def __init__(self):