Spaces:
Runtime error
Runtime error
MelikeDulkadir
commited on
Commit
•
b1dd89b
1
Parent(s):
38fafc9
Update app.py
Browse files
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()
|