Update app.py

app.py

@@ -51,11 +51,7 @@ def predict(image = None) :
     model.to(device)
 
     predicted = inference_fn(model, image)
-    
-    del model
-    gc.collect()
-    torch.cuda.empty_cache()
-    
+  
     return {labels[i]: float(predicted[i]) for i in range(10)}