m

app.py

@@ -1,4 +1,4 @@
-from hug_repository.sentiment_wrapper import PredictionModel
+from sentiment_wrapper import PredictionModel
 import gradio as gr
 
 model = PredictionModel()

sentiment_wrapper.py

@@ -2,7 +2,7 @@ from transformers import BertModel, BertTokenizer, AdamW, get_linear_schedule_wi
 from sklearn.metrics import  classification_report, f1_score
 from torch.utils.data import Dataset, DataLoader
 from tqdm.auto import tqdm
-from norbench_SA.config import params
+from config import params
 from torch import nn
 import pandas as pd
 import numpy as np