Upload app.py

app.py

@@ -1,25 +1,22 @@
-from model import DepressionClassifier
-
 import gradio as gr
 import numpy as np
-#from PIL import Image
-#import requests
+from transformers import pipeline
 
 import hopsworks
 import joblib
-import torch
+pipe = pipeline(model="liangc40/sentimental_analysis")
 
-project = hopsworks.login(project='liangc40')
+#project = hopsworks.login(project='liangc40')
 #fs = project.get_feature_store()
 
-mr = project.get_model_registry()
-model = mr.get_model("sentimental_analysis_model", version=3)
-model_dir = model.download()
-model = joblib.load(model_dir + "/sentimental_analysis_model.pkl")
-#model = torch.load(model_dir + "/sentimental_analysis_model.pkl", map_location=torch.device('cpu'))
+
+#mr = project.get_model_registry()
+#model = mr.get_model("sentimental_analysis_model", version=1)
+#model_dir = model.download()
+#model = joblib.load(model_dir + "/sentimental_analysis_model.pkl")
 
 def analyse(text):
-    label = model(text)
+    label = pipe(text)
     return label
 
 with gr.Blocks() as demo: