Spaces:

prithivMLmods
/

Imgscope-OCR-Mini

Running on Zero

prithivMLmods commited on Apr 9

Commit

a4b2163

verified ·

1 Parent(s): b282807

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 # Load text-only model and tokenizer (Pocket Llama)
 #model_id = "prithivMLmods/Pocket-Llama2-3.2-3B-Instruct"
-model_id = "prithivMLmods/Seyfert-Coder-14B-Y.1"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(

 # Load text-only model and tokenizer (Pocket Llama)
 #model_id = "prithivMLmods/Pocket-Llama2-3.2-3B-Instruct"
+model_id = "prithivMLmods/Venatici-Coder-14B-Y.2"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(