Fahad Ebrahim commited on
Commit
1a4bdfa
·
1 Parent(s): 5ac1d52
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def text_classification(text):
20
  model = AutoAdapterModel.from_pretrained("FacebookAI/roberta-base")
21
  adapter_react = model.load_adapter("buelfhood/irc-facebook-react", source = "hf",set_active=True)
22
  classifier = TextClassificationPipeline(model=model, tokenizer=tokenizer, max_length=256, padding="max_length", truncation=True,top_k=None)
23
- preprocessed_issue = preprocess (issue)
24
  out = classifier(preprocessed_issue)[0]
25
  return out
26
 
 
20
  model = AutoAdapterModel.from_pretrained("FacebookAI/roberta-base")
21
  adapter_react = model.load_adapter("buelfhood/irc-facebook-react", source = "hf",set_active=True)
22
  classifier = TextClassificationPipeline(model=model, tokenizer=tokenizer, max_length=256, padding="max_length", truncation=True,top_k=None)
23
+ preprocessed_issue = preprocess (text)
24
  out = classifier(preprocessed_issue)[0]
25
  return out
26