Spaces:
Sleeping
Sleeping
Commit
·
fdf5004
1
Parent(s):
c471abc
temppath
Browse files
app.py
CHANGED
|
@@ -86,7 +86,7 @@ async def predict_disease(file: UploadFile = File(...)):
|
|
| 86 |
# Validate file type
|
| 87 |
|
| 88 |
# Save uploaded file temporarily
|
| 89 |
-
with tempfile.NamedTemporaryFile(suffix=".jpg", delete=
|
| 90 |
temp_path = tmp.name
|
| 91 |
tmp.write(await file.read())
|
| 92 |
tmp.flush() # Ensure data is written
|
|
@@ -108,6 +108,8 @@ async def predict_disease(file: UploadFile = File(...)):
|
|
| 108 |
confidence = prediction[0][result_index]
|
| 109 |
disease_name = class_name[result_index]
|
| 110 |
|
|
|
|
|
|
|
| 111 |
return {
|
| 112 |
"success": True,
|
| 113 |
"disease": disease_name,
|
|
|
|
| 86 |
# Validate file type
|
| 87 |
|
| 88 |
# Save uploaded file temporarily
|
| 89 |
+
with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
|
| 90 |
temp_path = tmp.name
|
| 91 |
tmp.write(await file.read())
|
| 92 |
tmp.flush() # Ensure data is written
|
|
|
|
| 108 |
confidence = prediction[0][result_index]
|
| 109 |
disease_name = class_name[result_index]
|
| 110 |
|
| 111 |
+
os.remove(temp_path)
|
| 112 |
+
|
| 113 |
return {
|
| 114 |
"success": True,
|
| 115 |
"disease": disease_name,
|