Update app.py

app.py

@@ -75,7 +75,7 @@ def predict(image):
     with torch.no_grad():
         img_emb = model2.encode_image(inputs)
         img_emb = normalized(img_emb.cpu().numpy())
-        prediction = model(torch.from_numpy(img_emb).to(device).type(torch.cuda.FloatTensor)).item()
+        prediction = model(torch.from_numpy(img_emb).to(device).type(torch.FloatTensor)).item()
 
     result = {
         "clip_aesthetic": prediction,