From 93bcb8d51dab701030367b3b2f00f3976b3890c7 Mon Sep 17 00:00:00 2001
From: Yih-Dar SHIEH <chiapas@gmail.com>
Date: Sun, 24 Oct 2021 17:30:16 +0000
Subject: [PATCH] Update pipeline.py

---
 pipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pipeline.py b/pipeline.py
index cd5543e..2b7fef8 100644
--- a/pipeline.py
+++ b/pipeline.py
@@ -9,7 +9,7 @@ class PreTrainedPipeline():
 
     def __init__(self, path=""):
 
-        model_dir = os.path.join(path, "ckpt_epoch_3_step_6900")
+        model_dir = path
 
         self.model = FlaxVisionEncoderDecoderModel.from_pretrained(model_dir)
         self.feature_extractor = ViTFeatureExtractor.from_pretrained(model_dir)