Camille commited on
Commit
ef1f66e
·
1 Parent(s): 8384a73

fix: torch added

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import copy
2
  import logging
3
  from typing import List
4
 
 
5
  import streamlit as st
6
  from transformers import BertTokenizer, TFAutoModelForMaskedLM
7
  from transformers import CamembertModel, CamembertTokenizer
 
2
  import logging
3
  from typing import List
4
 
5
+ import torch
6
  import streamlit as st
7
  from transformers import BertTokenizer, TFAutoModelForMaskedLM
8
  from transformers import CamembertModel, CamembertTokenizer
requirements.txt CHANGED
@@ -2,4 +2,5 @@ gazpacho
2
  numpy
3
  requests
4
  tensorflow
5
- transformers
 
 
2
  numpy
3
  requests
4
  tensorflow
5
+ transformers
6
+ torch