Hugging Face's logo

Spaces:
balakrish181
/
cat_dog_classification
Sleeping

App Files Community
balakrish181 commited on
Commit
6388886
•
1 Parent(s): d957918

resolved issues

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
 
2
  import gradio as gr
3
  from model import model_classification
 
4
 
5
  path = 'efficient_cat_dog.pth'
6
  class_names = ['cat','dog']
 
1
 
2
  import gradio as gr
3
  from model import model_classification
4
+ import torch,os
5
 
6
  path = 'efficient_cat_dog.pth'
7
  class_names = ['cat','dog']