Update app.py

app.py

@@ -7,6 +7,7 @@ import cv2
 from PIL import Image
 from model import create_model
 model,transform=create_model(num_of_classes=3)
+model.load_state_dict(torch.load("fire_smoke_weight.pth", map_location=torch.device("cpu")))
 model.eval()
 def classify_video(video):
     cap = cv2.VideoCapture(video)