Commit
•
60b09de
1
Parent(s):
5b13338
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import AutoModelForSequenceClassification, AutoModel, AutoTokenizer
|
3 |
import torch
|
|
|
4 |
|
5 |
base_model = "cardiffnlp/twitter-roberta-base-sentiment-latest"
|
6 |
adapter_model = 'saideep-arikontham/twitter-roberta-base-sentiment-latest-trump-stance'
|
@@ -21,6 +22,5 @@ def greet(text):
|
|
21 |
return "This text is " + id2label[predictions.tolist()] + "!!"
|
22 |
|
23 |
|
24 |
-
|
25 |
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
|
26 |
demo.launch()
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import AutoModelForSequenceClassification, AutoModel, 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'
|
|
|
22 |
return "This text is " + id2label[predictions.tolist()] + "!!"
|
23 |
|
24 |
|
|
|
25 |
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
|
26 |
demo.launch()
|