Fahad Ebrahim
commited on
Commit
·
1a4bdfa
1
Parent(s):
5ac1d52
Fix bug
Browse files
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 (
|
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 |
|