Matt C commited on
Commit
632c54c
1 Parent(s): 300f06d

add pytorch?

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
-
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
4
 
5
 
6
  txt = st.text_area('Text to analyze', '''
 
1
  import streamlit as st
 
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
+ import torch
4
 
5
 
6
  txt = st.text_area('Text to analyze', '''
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- transformers
 
 
1
+ transformers
2
+ torch