Update main.py

main.py

@@ -27,7 +27,8 @@ async def upload_file(file: UploadFile = File(...), version: str = Form(...), sc
         temp_file_path = temp_file.name
 
     try:
-        result = client.predict(temp_file_path, api_name="/predict")
+        result = client.predict(temp_file_path, version, scale, api_name="/predict")
+
         
         # Check if the result is valid
         if result and len(result) == 2: