saideep-arikontham commited on
Commit
60a307f
1 Parent(s): 7fe75d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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")