Remove `torch` from import

app.py

@@ -1,10 +1,9 @@
 # Importing some modules
 import gradio as gr
 from transformers import pipeline
-import torch.cuda as cuda
 
 # Loading in the model
-MODEL = pipeline('image-classification', model='nateraw/vit-age-classifier', device=0 if cuda.is_available() else -1)
+MODEL = pipeline('image-classification', model='nateraw/vit-age-classifier', device=-1)
 
 # Main function to classify image
 def classify_image(image, top_k):