Update app.py

app.py

@@ -1,6 +1,3 @@
-import os
-os.system("pip install gradio==2.9b23")
-
 from pathlib import Path
 
 import torch
@@ -29,7 +26,8 @@ state_dict = torch.load('pytorch_model.bin', map_location='cpu')
 model.load_state_dict(state_dict, strict=False)
 model.eval()
 
-def predict(im):
+def predict(input):
+    im = input
     if im is None:
         return None