missing library and incorrect paths

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from PIL import Image
 import numpy as np
+import torch
+from torchvision import datasets, transforms
+
 
 # Load the saved model
-model = torch.load('/kaggle/input/models/model.pth', map_location=torch.device('cpu'))
+model = torch.load('./model.pth', map_location=torch.device('cpu'))
 model.eval()
 
 # Define the transformation for preprocessing
@@ -42,15 +45,11 @@ def predict_sketch(image):
 	# Make the prediction
 	with torch.no_grad():
 		output = model(image)
-		print("output", output)
 		probabilities = torch.nn.functional.softmax(output[0], dim=0)
-		print("prob" , probabilities)
 
 	labels = [str(i) for i in range(10)]
 	probabilities = probabilities.cpu().numpy()
 	result = {label: float(prob) for label, prob in zip(labels, probabilities)}
-	print("probabilities: ", probabilities)
-	print("result: ", result)
 
 	return result