Spaces:

lyrisha
/

sentiment-analysis

Running

lyrisha commited on Apr 9, 2024

Commit

e4464db

1 Parent(s): 6e02732

Fix app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 class UI:
-    def __init__(self, model):
         self.model = pipeline("sentiment-analysis", model="lyrisha/bert-finetuned-imdb-sentiment")
     def launch(self):

 from transformers import pipeline
 class UI:
+    def __init__(self):
         self.model = pipeline("sentiment-analysis", model="lyrisha/bert-finetuned-imdb-sentiment")
     def launch(self):