Update app.py

app.py

@@ -85,7 +85,7 @@ class TorchPredictWrapper:
 def load_model(file_path: str):
     try:
         model = joblib.load(file_path)
-        return SKLearnWrapper(model)
+        return SKLearnWrapper(model)  # now wrapping just clf
     except Exception as e1:
         try:
             with open(file_path, "rb") as f: