zeroix07 commited on
Commit
32c6bcd
1 Parent(s): 4962875

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -6
app.py CHANGED
@@ -1,5 +1,3 @@
1
- pip install setfit
2
-
3
  import gradio as gr
4
  from setfit import AbsaModel
5
 
@@ -10,11 +8,19 @@ examples = [["Harganya sangat masuk akal untuk makanan pembuka dan makanannya ya
10
  model = AbsaModel.from_pretrained(
11
  "zeroix07/indo-setfit-absa-model-aspect",
12
  "zeroix07/indo-setfit-absa-model-polarity",
 
13
  )
14
-
15
- interface = gr.Interface.load(model,
16
- description=description,
17
- examples=examples
 
 
 
 
 
 
 
18
  )
19
 
20
  interface.launch()
 
 
 
1
  import gradio as gr
2
  from setfit import AbsaModel
3
 
 
8
  model = AbsaModel.from_pretrained(
9
  "zeroix07/indo-setfit-absa-model-aspect",
10
  "zeroix07/indo-setfit-absa-model-polarity",
11
+ spacy_model="id_core_news_trf",
12
  )
13
+ def analyze_text(text):
14
+ pred = model.predict(text)
15
+ return pred
16
+
17
+ interface = gr.Interface(
18
+ fn=analyze_text,
19
+ inputs="textbox",
20
+ outputs="text",
21
+ description=description,
22
+ title=title,
23
+ examples=examples
24
  )
25
 
26
  interface.launch()