yavuzkomecoglu commited on
Commit
b9bec37
1 Parent(s): 1d2f10c

refactor results

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -10,11 +10,15 @@ model = AutoModelForSequenceClassification.from_pretrained("serdarakyol/interpre
10
 
11
  def prediction(news):
12
  # create pipeline
13
- clasifer = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model)
14
 
15
- pred = clasifer(news)
16
- return str(pred[0]["label"]), str(round(pred[0]["score"], 5))
17
 
 
 
 
 
 
18
 
19
 
20
  gradio_ui = gr.Interface(
@@ -27,11 +31,8 @@ gradio_ui = gr.Interface(
27
  ["Instagram, 3 yıl önce kullanıcıların beğenisine sunduğu IGTV özelliğini kaldırıyor. IGTV’nin yerine gelecek yeni özelliğin ise platformun Reels'tan elde ettiği başarıyı pekiştirip içerik üreticileri için kullanışlı bir zemin oluşturması bekleniyor."],
28
  ["Kamu bankaları Halkbank, Vakıfbank ve Ziraat Bankası ortak bir açıklama yayınlayarak kredi faiz oranlarında indirim yapıldığını açıkladı. Söz konusu indirimler ticari ve konut kredileri kapsıyor. Tüketici kredisi kapsam dışında bırakıldı."]
29
  ],
30
- inputs=gr.inputs.Textbox(lines=10, label="Paste some text here"),
31
- outputs=[
32
- gr.outputs.Textbox(label="News Category Label"),
33
- gr.outputs.Textbox(label="News Category Score"),
34
- ],
35
  theme="huggingface",
36
  article="<p style='text-align: center'>for our other AI works: <a href='https://www.kodiks.com/ai_solutions.html' target='_blank'>https://www.kodiks.com/ai_solutions.html</a> | <a href='https://twitter.com/KodiksBilisim' target='_blank'>Contact us</a></p>",
37
  )
 
10
 
11
  def prediction(news):
12
  # create pipeline
13
+ clasifer = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model, return_all_scores=True)
14
 
15
+ preds = clasifer(news)
 
16
 
17
+ preds_dict={}
18
+ for pred in preds[0]:
19
+ preds_dict[pred['label']] = pred['score']
20
+
21
+ return preds_dict
22
 
23
 
24
  gradio_ui = gr.Interface(
 
31
  ["Instagram, 3 yıl önce kullanıcıların beğenisine sunduğu IGTV özelliğini kaldırıyor. IGTV’nin yerine gelecek yeni özelliğin ise platformun Reels'tan elde ettiği başarıyı pekiştirip içerik üreticileri için kullanışlı bir zemin oluşturması bekleniyor."],
32
  ["Kamu bankaları Halkbank, Vakıfbank ve Ziraat Bankası ortak bir açıklama yayınlayarak kredi faiz oranlarında indirim yapıldığını açıkladı. Söz konusu indirimler ticari ve konut kredileri kapsıyor. Tüketici kredisi kapsam dışında bırakıldı."]
33
  ],
34
+ inputs=gr.inputs.Textbox(lines=10, label="Paste some news here"),
35
+ outputs=gr.outputs.Label(num_top_classes=5, type="auto", label="News categories probabilities"),
 
 
 
36
  theme="huggingface",
37
  article="<p style='text-align: center'>for our other AI works: <a href='https://www.kodiks.com/ai_solutions.html' target='_blank'>https://www.kodiks.com/ai_solutions.html</a> | <a href='https://twitter.com/KodiksBilisim' target='_blank'>Contact us</a></p>",
38
  )