Spaces:
Runtime error
Runtime error
edit
Browse files- app.py +2 -0
- requirements.txt +2 -1
app.py
CHANGED
@@ -1,11 +1,13 @@
|
|
1 |
import numpy as np
|
2 |
import streamlit as st
|
3 |
from transformers import pipeline
|
|
|
4 |
|
5 |
def bertweet(data):
|
6 |
specific_model = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
|
7 |
print(specific_model(data))
|
8 |
|
|
|
9 |
def getSent(data, model):
|
10 |
if(model == 'Bertweet'):
|
11 |
bertweet(data)
|
|
|
1 |
import numpy as np
|
2 |
import streamlit as st
|
3 |
from transformers import pipeline
|
4 |
+
import torch
|
5 |
|
6 |
def bertweet(data):
|
7 |
specific_model = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
|
8 |
print(specific_model(data))
|
9 |
|
10 |
+
|
11 |
def getSent(data, model):
|
12 |
if(model == 'Bertweet'):
|
13 |
bertweet(data)
|
requirements.txt
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
numpy
|
2 |
streamlit
|
3 |
transformers
|
4 |
-
tensorflow
|
|
|
|
1 |
numpy
|
2 |
streamlit
|
3 |
transformers
|
4 |
+
tensorflow
|
5 |
+
torch
|