dahongj commited on
Commit
c45446a
1 Parent(s): 728a13e
Files changed (2) hide show
  1. app.py +2 -0
  2. 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