Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ repo_id = "JeanCGuerrero/Practica2"
|
|
| 10 |
model = AutoModelForObjectDetection.from_pretrained(repo_id)
|
| 11 |
image_processor = AutoImageProcessor.from_pretrained(repo_id)
|
| 12 |
|
| 13 |
-
# Funci贸n
|
| 14 |
def predict(img):
|
| 15 |
img = img.convert("RGB") # Asegurar que la imagen est茅 en formato RGB
|
| 16 |
inputs = image_processor(images=img, return_tensors="pt")
|
|
@@ -45,3 +45,4 @@ gr.Interface(
|
|
| 45 |
description="Este modelo detecta mapaches en im谩genes usando DETR con el dataset Raccoon.",
|
| 46 |
examples=['raccoon-108.jpg', 'raccoon-108.jpg']
|
| 47 |
).launch(share=False)
|
|
|
|
|
|
| 10 |
model = AutoModelForObjectDetection.from_pretrained(repo_id)
|
| 11 |
image_processor = AutoImageProcessor.from_pretrained(repo_id)
|
| 12 |
|
| 13 |
+
# Funci贸n para la inferencia
|
| 14 |
def predict(img):
|
| 15 |
img = img.convert("RGB") # Asegurar que la imagen est茅 en formato RGB
|
| 16 |
inputs = image_processor(images=img, return_tensors="pt")
|
|
|
|
| 45 |
description="Este modelo detecta mapaches en im谩genes usando DETR con el dataset Raccoon.",
|
| 46 |
examples=['raccoon-108.jpg', 'raccoon-108.jpg']
|
| 47 |
).launch(share=False)
|
| 48 |
+
|