Portx
/

do_extractor_test

Model card Files Files and versions

Portx commited on Mar 6, 2025

Commit

d914a4b

·

verified ·

1 Parent(s): 66b6e5b

Update handler.py

Files changed (1) hide show

handler.py +0 -1

handler.py CHANGED Viewed

@@ -9,7 +9,6 @@ run("pip install torch torchvision torchaudio --extra-index-url https://download
 #run("pip uninstall unsloth -y && pip install --upgrade --no-cache-dir --no-deps git+https://github.com/unslothai/unsloth.git", shell=True, check=True)
 import torch
-from unsloth.trainer import UnslothVisionDataCollator
 from unsloth import FastVisionModel
 from PIL import Image
 import re

 #run("pip uninstall unsloth -y && pip install --upgrade --no-cache-dir --no-deps git+https://github.com/unslothai/unsloth.git", shell=True, check=True)
 import torch
 from unsloth import FastVisionModel
 from PIL import Image
 import re