Update app.py

app.py

@@ -8,6 +8,14 @@ from PIL import Image
 from transformers import CLIPModel, CLIPProcessor
 import gradio as gr
 
+# Ensure PyTorch is installed
+try:
+    import torch
+except ModuleNotFoundError:
+    print("PyTorch is not installed. Installing now...")
+    os.system("pip install torch torchvision torchaudio")
+    import torch
+
 # Step 1: Unzip the dataset
 if not os.path.exists("data"):
     os.makedirs("data")