Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -27,7 +27,8 @@ async def upload_file(file: UploadFile = File(...), version: str = Form(...), sc
|
|
27 |
temp_file_path = temp_file.name
|
28 |
|
29 |
try:
|
30 |
-
result = client.predict(temp_file_path, api_name="/predict")
|
|
|
31 |
|
32 |
# Check if the result is valid
|
33 |
if result and len(result) == 2:
|
|
|
27 |
temp_file_path = temp_file.name
|
28 |
|
29 |
try:
|
30 |
+
result = client.predict(temp_file_path, version, scale, api_name="/predict")
|
31 |
+
|
32 |
|
33 |
# Check if the result is valid
|
34 |
if result and len(result) == 2:
|