Spaces:

knowledgator
/

GLiClass_SandBox

Running

BioMike commited on Jul 22

Commit

39eb373

verified ·

1 Parent(s): f3f51fa

Update interfaces/main_pipeline.py

Files changed (1) hide show

interfaces/main_pipeline.py CHANGED Viewed

@@ -7,7 +7,7 @@ from gliclass import GLiClassModel, ZeroShotClassificationPipeline
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
-model_path = "knowledgator/gliclass-edge-v3.0"
 model = GLiClassModel.from_pretrained(model_path)
 tokenizer = AutoTokenizer.from_pretrained(model_path)

 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
+model_path = "knowledgator/gliclass-base-v1.0-lw"
 model = GLiClassModel.from_pretrained(model_path)
 tokenizer = AutoTokenizer.from_pretrained(model_path)