201912103_furniture_use_data_finetuning

# Carga el modelo y realiza predicciones. 
from transformers import DetrForObjectDetection, DetrImageProcessor 
import torch 
from PIL import Image 
 
# Cargar el modelo 
model = DetrForObjectDetection.from_pretrained('timmy0079/201912103_furniture_use_data_finetuning') 
processor = DetrImageProcessor.from_pretrained('timmy0079/201912103_furniture_use_data_finetuning') 
 
# Abrir la imagen y realizar procesamiento 
image = Image.open('path_to_your_image.png') 
inputs = processor(images=image, return_tensors='pt') 
 
# Realizar predicción 
with torch.no_grad(): 
    outputs = model(**inputs) 
 
# Extraer la información de la predicción 
predictions = outputs['logits'].softmax(-1).argmax(-1) 
print(predictions)