Update app.py

app.py

@@ -1,4 +1,10 @@
-import torch
+try:
+    import torch
+except ImportError:
+    import os
+    os.system("pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu")
+    import torch
+
 import torchvision.transforms as transforms
 from torchvision.models import resnet18, ResNet18_Weights
 from PIL import Image
@@ -7,6 +13,7 @@ import gradio as gr
 # Print Gradio version
 print("Gradio version:", gr.__version__)
 
+
 # Load Pretrained ResNet18 Model
 weights = ResNet18_Weights.DEFAULT
 model = resnet18(weights=weights)