HinaCortus commited on
Commit
bd190f2
·
1 Parent(s): ed2c942

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ def server(input: Inputs, output: Outputs, session: Session):
18
  loaded_model = AutoModelForSequenceClassification.from_pretrained(model_name)
19
 
20
  nlp = TextClassificationPipeline(model=loaded_model, tokenizer=loaded_tokenizer)
21
- nlp(Inputs, truncation=True)
22
-
23
 
24
 
25
  app = App(
 
18
  loaded_model = AutoModelForSequenceClassification.from_pretrained(model_name)
19
 
20
  nlp = TextClassificationPipeline(model=loaded_model, tokenizer=loaded_tokenizer)
21
+ result = nlp(Inputs, truncation=True)
22
+ return result
23
 
24
 
25
  app = App(