File size: 737 Bytes
d58d93d c06b09c d58d93d c06b09c d58d93d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from fastapi import FastAPI
from fastapi.responses import JSONResponse
import requests
from PIL import Image
import torch
from io import BytesIO
app = FastAPI()
# Carga el modelo de detecci贸n (puedes cambiarlo seg煤n tu necesidad)
model = torch.hub.load('facebook/detr', 'detr_resnet50', pretrained=True)
model.eval()
@app.get("/predict/")
async def predict(url: str):
response = requests.get(url)
img = Image.open(BytesIO(response.content)).convert("RGB")
# Aqu铆 deber铆as incluir el preprocesamiento de la imagen y la l贸gica para la detecci贸n.
# Por ahora, simulamos la detecci贸n.
detected = True # Cambia esto seg煤n la l贸gica de detecci贸n
return JSONResponse(content={"detected": int(detected)})
|