Update app.py

app.py

@@ -1,8 +1,17 @@
 import os
-if not os.getenv("UNSLOTH_INSTALLED"):
-    os.system('pip install git+https://github.com/unslothai/unsloth.git')
-    os.environ["UNSLOTH_INSTALLED"] = "1"
 
+# Install PyTorch if it's missing
+try:
+    import torch
+except ImportError:
+    os.system("pip install torch")
+
+# Install Unsloth if missing
+try:
+    from unsloth import FastVisionModel
+except ImportError:
+    os.system('pip install git+https://github.com/unslothai/unsloth.git')
+    from unsloth import FastVisionModel
 
 
 import gradio as gr