momo commited on
Commit
e24f20d
1 Parent(s): 65aabc9
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  """
2
  python interactive.py
3
  """
 
4
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
  from transformers import TextClassificationPipeline
6
  import gradio as gr
 
1
  """
2
  python interactive.py
3
  """
4
+ import torch
5
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
6
  from transformers import TextClassificationPipeline
7
  import gradio as gr
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- transformers==4.5.0
 
 
1
+ transformers==4.5.0
2
+ torch==1.9.0