databoyface commited on
Commit
584b8a3
·
verified ·
1 Parent(s): 03d8f7f

Renamed input

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import pipeline
3
 
4
  pipeline = pipeline(task="text-classification", model="databoyface/distilbert-base-cased-ome-v4.2")
5
 
6
- def predict(input_txt):
7
- predictions = pipeline(input_txt)
8
  return {p["label"]: p["score"] for p in predictions}
9
 
10
  gradio_app = gr.Interface(
 
3
 
4
  pipeline = pipeline(task="text-classification", model="databoyface/distilbert-base-cased-ome-v4.2")
5
 
6
+ def predict(input):
7
+ predictions = pipeline(input)
8
  return {p["label"]: p["score"] for p in predictions}
9
 
10
  gradio_app = gr.Interface(