Update app.py

app.py

@@ -1,28 +1,27 @@
 import subprocess
 import sys
 
-# Function to install a package
+# Ensure required packages are installed
 def install(package):
     subprocess.check_call([sys.executable, "-m", "pip", "install", package])
 
-# Ensure required libraries are installed
 try:
-    import torch
+    import numpy as np
 except ImportError:
-    install("torch==2.0.1")
-    install("torchvision==0.15.2")
+    install("numpy<2")
 
 try:
-    import numpy as np
+    import torch
 except ImportError:
-    install("numpy<2")
+    install("torch==2.0.1")
+    install("torchvision==0.15.2")
 
 try:
     from PIL import Image
 except ImportError:
     install("Pillow==9.5.0")
 
-# Imports
+# Import necessary libraries
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -30,7 +29,6 @@ import torchvision.transforms as transforms
 from PIL import Image
 import gradio as gr
 
-
 # Define the model
 class ModifiedLargeNet(nn.Module):
     def __init__(self):
@@ -50,7 +48,6 @@ class ModifiedLargeNet(nn.Module):
         x = self.fc2(x)
         return x
 
-
 # Load the trained model
 model = ModifiedLargeNet()
 model.load_state_dict(torch.load("modified_large_net.pt", map_location=torch.device("cpu")))