Hugging Face's logo

Spaces:
subhranil2699
/
shape_classifier
Runtime error

App Files Community
subhranil2699 commited on
Commit
86dca36
·
1 Parent(s): fad8c25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,7 +4,10 @@ import tensorflow as tf
4
  import numpy as np
5
 
6
 
7
- MODEL = tf.keras.models.load_model("my_model")
 
 
 
8
  CLASS_NAMES = ['circle', 'rectangle']
9
 
10
 
 
4
  import numpy as np
5
 
6
 
7
+ from huggingface_hub import from_pretrained_keras
8
+
9
+ MODEL = from_pretrained_keras("subhranil2699/shape_classifier_tf")
10
+
11
  CLASS_NAMES = ['circle', 'rectangle']
12
 
13