Spaces:
Sleeping
Sleeping
muhammadhamza-stack
commited on
Commit
·
d3519ad
1
Parent(s):
f23f0f3
change image type to pil
Browse files- .gitattributes +1 -0
- .gitignore +1 -0
- app.py +1 -4
- requirements.txt +3 -1
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
*.jpg filter=lfs diff=lfs merge=lfs -text
|
.gitignore
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
venv
|
app.py
CHANGED
|
@@ -65,8 +65,6 @@ OUTPUT_EXPLANATION = """
|
|
| 65 |
"""
|
| 66 |
|
| 67 |
# --- Example Data Setup ---
|
| 68 |
-
# Assuming you have example images named 'blood_smear_1.jpg' and 'blood_smear_2.jpg'
|
| 69 |
-
# in an 'examples' folder or the root directory.
|
| 70 |
SAMPLE_EXAMPLES = [
|
| 71 |
["blood_smear_1.jpg", "Yolo V11", 0.5],
|
| 72 |
["blood_smear_2.jpg", "Real Time Detection Transformer", 0.45],
|
|
@@ -75,7 +73,6 @@ SAMPLE_EXAMPLES = [
|
|
| 75 |
# ----------------- Core Inference Function -----------------
|
| 76 |
|
| 77 |
def inference(image, model, conf) -> Tuple[str, str, str]:
|
| 78 |
-
# Ensure all inputs are valid before proceeding
|
| 79 |
if image is None:
|
| 80 |
gr.Error("Please upload an image.")
|
| 81 |
return None, "0", "0"
|
|
@@ -144,7 +141,7 @@ with gr.Blocks(title="Blood Cell Detection") as ifer:
|
|
| 144 |
with gr.Row():
|
| 145 |
with gr.Column():
|
| 146 |
gr.Markdown("## Step 1: Upload Image ")
|
| 147 |
-
image_input = gr.Image(label="Input Image", type="
|
| 148 |
with gr.Column():
|
| 149 |
gr.Markdown("## Step 2: Set Parameters")
|
| 150 |
model_selection = gr.Dropdown(
|
|
|
|
| 65 |
"""
|
| 66 |
|
| 67 |
# --- Example Data Setup ---
|
|
|
|
|
|
|
| 68 |
SAMPLE_EXAMPLES = [
|
| 69 |
["blood_smear_1.jpg", "Yolo V11", 0.5],
|
| 70 |
["blood_smear_2.jpg", "Real Time Detection Transformer", 0.45],
|
|
|
|
| 73 |
# ----------------- Core Inference Function -----------------
|
| 74 |
|
| 75 |
def inference(image, model, conf) -> Tuple[str, str, str]:
|
|
|
|
| 76 |
if image is None:
|
| 77 |
gr.Error("Please upload an image.")
|
| 78 |
return None, "0", "0"
|
|
|
|
| 141 |
with gr.Row():
|
| 142 |
with gr.Column():
|
| 143 |
gr.Markdown("## Step 1: Upload Image ")
|
| 144 |
+
image_input = gr.Image(label="Input Image", type="pil")
|
| 145 |
with gr.Column():
|
| 146 |
gr.Markdown("## Step 2: Set Parameters")
|
| 147 |
model_selection = gr.Dropdown(
|
requirements.txt
CHANGED
|
@@ -1,2 +1,4 @@
|
|
| 1 |
ultralytics
|
| 2 |
-
|
|
|
|
|
|
|
|
|
| 1 |
ultralytics
|
| 2 |
+
numpy<2
|
| 3 |
+
gradio==3.50.2
|
| 4 |
+
gradio-client==0.6.1
|