JeanCGuerrero commited on
Commit
270075a
verified
1 Parent(s): a158c93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 que se encarga de llevar a cabo las predicciones
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
+