Spaces:

syedfaisalabrar
/

License_Classification

Sleeping

syedfaisalabrar commited on Feb 26

Commit

18462ef

verified ·

1 Parent(s): 383aa7e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ model_path = "best.pt"
 modelY = YOLO(model_path)
 os.environ["TRANSFORMERS_CACHE"] = "./.cache"
 cache_folder = "./.cache"
 # Load the Hugging Face model and tokenizer globally (downloaded only once)
 model = AutoModel.from_pretrained(
     path,

 modelY = YOLO(model_path)
 os.environ["TRANSFORMERS_CACHE"] = "./.cache"
 cache_folder = "./.cache"
+path = "OpenGVLab/InternVL2_5-4B"
 # Load the Hugging Face model and tokenizer globally (downloaded only once)
 model = AutoModel.from_pretrained(
     path,