brendenc commited on
Commit
303db69
·
1 Parent(s): 945aef0

Updated from colab

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
3
  import torch
4
  import matplotlib.pyplot as plt
 
1
  import gradio as gr
2
+ import numpy as np
3
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
4
  import torch
5
  import matplotlib.pyplot as plt
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  torch
2
  transformers
3
  matplotlib
 
 
1
  torch
2
  transformers
3
  matplotlib
4
+ numpy