Updates

src/models/predict_model.py

@@ -12,7 +12,7 @@ def predict_model(text):
         params = yaml.safe_load(f)
 
     model = Summarization()
-    model.load_model(model_type=params["model_type"], model_dir=params["model_dir"])
+    model.load_model(model_type=params["model_type"], model_dir=f"{params['hf_username']}/{params['name']}")
     pre_summary = model.predict(text)
     return pre_summary
 

src/visualization/visualize.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import yaml
 
-from models import predict_model
+from src.models.predict_model import predict_model
 
 
 def visualize():