Spaces:

Pavan147
/

Docling_Image

Sleeping

Pavan147 commited on 6 days ago

Commit

ea5c0d6

verified ·

1 Parent(s): df46f51

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,7 @@ from PIL import Image, ImageOps
 # Load model & processor once at startup
 processor = AutoProcessor.from_pretrained("ds4sd/SmolDocling-256M-preview")
-model = AutoModelForVision2Seq.from_pretrained("ds4sd/SmolDocling-256M-preview", torch_dtype=torch.bfloat16).to("cuda")
 def add_random_padding(image, min_percent=0.1, max_percent=0.10):
     image = image.convert("RGB")

 # Load model & processor once at startup
 processor = AutoProcessor.from_pretrained("ds4sd/SmolDocling-256M-preview")
+model = AutoModelForVision2Seq.from_pretrained("ds4sd/SmolDocling-256M-preview")
 def add_random_padding(image, min_percent=0.1, max_percent=0.10):
     image = image.convert("RGB")