kataniccc commited on
Commit
c480dfd
1 Parent(s): 6f74060

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -1,8 +1,14 @@
1
  import gradio as gr
2
  import pandas as pd
3
 
 
 
 
4
  from datasets import Dataset,DatasetDict
5
 
 
 
 
6
  df = pd.read_csv(path/'train.csv')
7
 
8
  df.describe(include='object')
@@ -19,8 +25,10 @@ def predict_text(input_text):
19
  return prediction
20
 
21
 
 
22
  text_input = gr.inputs.Textbox(lines=7, label="Unesite tekst")
23
  output_text = gr.outputs.Textbox(label="Predikcija")
24
 
25
  gr.Interface(predict_text, inputs=text_input, outputs=output_text).launch()
26
 
 
 
1
  import gradio as gr
2
  import pandas as pd
3
 
4
+ from transformers import pipeline
5
+
6
+
7
  from datasets import Dataset,DatasetDict
8
 
9
+ classifier = pipeline("text-classification", model=model_nm)
10
+
11
+
12
  df = pd.read_csv(path/'train.csv')
13
 
14
  df.describe(include='object')
 
25
  return prediction
26
 
27
 
28
+
29
  text_input = gr.inputs.Textbox(lines=7, label="Unesite tekst")
30
  output_text = gr.outputs.Textbox(label="Predikcija")
31
 
32
  gr.Interface(predict_text, inputs=text_input, outputs=output_text).launch()
33
 
34
+