saideep-arikontham
commited on
Commit
•
60a307f
1
Parent(s):
7fe75d1
Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
|
|
2 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
3 |
import torch
|
4 |
from peft import PeftModel, PeftConfig
|
|
|
5 |
|
6 |
base_model = "cardiffnlp/twitter-roberta-base-sentiment-latest"
|
7 |
adapter_model = 'saideep-arikontham/twitter-roberta-base-sentiment-latest-trump-stance-1'
|
@@ -16,8 +17,9 @@ model = AutoModelForSequenceClassification.from_pretrained(base_model, num_label
|
|
16 |
model = PeftModel.from_pretrained(model, adapter_model)
|
17 |
tokenizer = AutoTokenizer.from_pretrained(adapter_model)
|
18 |
|
19 |
-
def greet(text):
|
20 |
|
|
|
|
|
21 |
|
22 |
model.to('cpu')
|
23 |
inputs = tokenizer.encode(text, return_tensors="pt").to("cpu")
|
|
|
2 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
3 |
import torch
|
4 |
from peft import PeftModel, PeftConfig
|
5 |
+
import re
|
6 |
|
7 |
base_model = "cardiffnlp/twitter-roberta-base-sentiment-latest"
|
8 |
adapter_model = 'saideep-arikontham/twitter-roberta-base-sentiment-latest-trump-stance-1'
|
|
|
17 |
model = PeftModel.from_pretrained(model, adapter_model)
|
18 |
tokenizer = AutoTokenizer.from_pretrained(adapter_model)
|
19 |
|
|
|
20 |
|
21 |
+
|
22 |
+
def greet(text):
|
23 |
|
24 |
model.to('cpu')
|
25 |
inputs = tokenizer.encode(text, return_tensors="pt").to("cpu")
|