MelikeDulkadir commited on
Commit
b1dd89b
1 Parent(s): 38fafc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
4
  from datasets import load_dataset
5
  imdb = load_dataset("imdb")
6
 
7
- from transformers import AutoTokenizer
8
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
9
 
10
  def preprocess_function(examples):
@@ -36,4 +36,11 @@ trainer = Trainer(
36
  data_collator=data_collator,
37
  )
38
 
39
- trainer.train()
 
 
 
 
 
 
 
 
4
  from datasets import load_dataset
5
  imdb = load_dataset("imdb")
6
 
7
+ from transformers import AutoTokenizer, pipeline
8
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
9
 
10
  def preprocess_function(examples):
 
36
  data_collator=data_collator,
37
  )
38
 
39
+ trainer.train()
40
+
41
+ def greet(text):
42
+ pipe = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model)
43
+ return pipe(text)[0]['label']
44
+
45
+ iface = gr.Interface(fn=greet, inputs=gr.inputs.Textbox(placeholder="Lütfen Cümle Giriniz...", lines=5), outputs="text")
46
+ iface.launch()