ossaili commited on
Commit
acbbc7d
1 Parent(s): b96f427
Files changed (2) hide show
  1. app.py +0 -4
  2. models.py +0 -7
app.py CHANGED
@@ -5,10 +5,6 @@ from models import efficientnetv2b0_25_arch_styles_Classifier
5
 
6
 
7
  demo = gr.Interface(
8
- <<<<<<< HEAD
9
  efficientnetv2b0_25_arch_styles_Classifier, "image", "label")
10
- =======
11
- efficientnetv2b0_25_arch_styles_Classifier, "image", outputs=["text"])
12
- >>>>>>> 6dfb3eb853ff6baf202a3117ebb5b61755454768
13
 
14
  demo.launch()
 
5
 
6
 
7
  demo = gr.Interface(
 
8
  efficientnetv2b0_25_arch_styles_Classifier, "image", "label")
 
 
 
9
 
10
  demo.launch()
models.py CHANGED
@@ -41,16 +41,9 @@ def efficientnetv2b0_25_arch_styles_Classifier(image):
41
  224, 224), interpolation=cv2.INTER_CUBIC)
42
  img = tf.expand_dims(resized_image, 0)
43
  efficientnetv2b0 = tf.keras.models.load_model(
44
- <<<<<<< HEAD
45
  "models\EfficientNetV2B0.h5")
46
 
47
  y = efficientnetv2b0.predict(img).reshape(-1)
48
  y = (np.round(y, 3)).tolist()
49
- =======
50
- "EfficientNetV2B0.h5")
51
-
52
- y = efficientnetv2b0.predict(img).reshape(-1)
53
- y = (np.round(y, 3)*100).tolist()
54
- >>>>>>> 6dfb3eb853ff6baf202a3117ebb5b61755454768
55
 
56
  return outputs(y)
 
41
  224, 224), interpolation=cv2.INTER_CUBIC)
42
  img = tf.expand_dims(resized_image, 0)
43
  efficientnetv2b0 = tf.keras.models.load_model(
 
44
  "models\EfficientNetV2B0.h5")
45
 
46
  y = efficientnetv2b0.predict(img).reshape(-1)
47
  y = (np.round(y, 3)).tolist()
 
 
 
 
 
 
48
 
49
  return outputs(y)