Update app.py

app.py

@@ -17,7 +17,7 @@ processor = TrOCRProcessor.from_pretrained('microsoft/trocr-large-handwritten')
 trocr_model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-large-handwritten').to(device)
 trocr_model.config.num_beams = 1
 
-yolo_model = YOLO(yolo_weights_path).to('mps')
+yolo_model = YOLO(yolo_weights_path).to('cpu')
 unmasker_large = pipeline('fill-mask', model='roberta-large', device=device)
 roberta_model = AutoModelForMaskedLM.from_pretrained("roberta-large").to(device)