marta-marta commited on
Commit
c4ca994
·
1 Parent(s): 149e201

Updating to get the model imports to work

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -8,7 +8,7 @@ import streamlit as st
8
  # from transformers import TFAutoModel
9
  # Needed for importing torch to use in the transformers model
10
  # import torch
11
- # import tensorflow
12
  import matplotlib.pyplot as plt
13
  # HELLO HUGGING FACE
14
 
 
8
  # from transformers import TFAutoModel
9
  # Needed for importing torch to use in the transformers model
10
  # import torch
11
+ import tensorflow
12
  import matplotlib.pyplot as plt
13
  # HELLO HUGGING FACE
14
 
requirements.txt CHANGED
@@ -3,3 +3,4 @@ matplotlib==3.5.2
3
  numpy==1.21.5
4
  scipy==1.9.1
5
  streamlit==1.18.1
 
 
3
  numpy==1.21.5
4
  scipy==1.9.1
5
  streamlit==1.18.1
6
+ tensorflow==2.10.0