Spaces:

saideep-arikontham
/

trump_stance_detector

Sleeping

saideep-arikontham commited on Apr 20

Commit

2f7bc0e

•

1 Parent(s): 88fd799

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,9 +2,12 @@ import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoModel, AutoTokenizer
 import torch
-model_checkpoint = 'saideep-arikontham/twitter-roberta-base-sentiment-latest-trump-stance'
-model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2, ignore_mismatched_sizes=True)
-tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 def greet(text):

 from transformers import AutoModelForSequenceClassification, AutoModel, AutoTokenizer
 import torch
+base_model = "cardiffnlp/twitter-roberta-base-sentiment-latest"
+adapter_model = 'saideep-arikontham/twitter-roberta-base-sentiment-latest-trump-stance'
+model = PeftModel.from_pretrained(model, adapter_model)
+tokenizer = AutoTokenizer.from_pretrained(adapter_model)
 def greet(text):