Spaces:

stamps-labs
/

stamp2vec

Runtime error

sadjava commited on Jul 16, 2023

Commit

10618a1

1 Parent(s): 1ec96d4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ from pipelines.feature_extraction.vits8 import Vits8Pipeline
 from utils import *
-yolov8 = Yolov8Pipeline.from_pretrained(local_model_path='yolov8_old_backup.pt')
 yolo_stamp = YoloStampPipeline.from_pretrained('stamps-labs/yolo-stamp', 'weights.pt')
 vae = VaePipeline.from_pretrained('stamps-labs/vae-encoder', 'weights.pt')
 vits8 = Vits8Pipeline.from_pretrained('stamps-labs/vits8-stamp', 'weights.pt')

 from utils import *
+yolov8 = Yolov8Pipeline.from_pretrained('stamps-labs/yolov8-finetuned', 'weights.pt')
 yolo_stamp = YoloStampPipeline.from_pretrained('stamps-labs/yolo-stamp', 'weights.pt')
 vae = VaePipeline.from_pretrained('stamps-labs/vae-encoder', 'weights.pt')
 vits8 = Vits8Pipeline.from_pretrained('stamps-labs/vits8-stamp', 'weights.pt')