Update app.py
Browse files
app.py
CHANGED
|
@@ -135,7 +135,7 @@ def ocr(image):
|
|
| 135 |
pixel_values = processor(image, return_tensors='pt').pixel_values.to(device)
|
| 136 |
generated_ids = trained_model.generate(pixel_values)
|
| 137 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 138 |
-
return
|
| 139 |
except Exception as e:
|
| 140 |
return f"Erreur lors du traitement de l'image : {e}"
|
| 141 |
|
|
|
|
| 135 |
pixel_values = processor(image, return_tensors='pt').pixel_values.to(device)
|
| 136 |
generated_ids = trained_model.generate(pixel_values)
|
| 137 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 138 |
+
return generated_text
|
| 139 |
except Exception as e:
|
| 140 |
return f"Erreur lors du traitement de l'image : {e}"
|
| 141 |
|