Zahraebrahimi commited on
Commit
6833680
·
1 Parent(s): 7a7a70e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -2,9 +2,6 @@ import gradio as gr
2
  import argparse
3
  import numpy as np
4
  import pickle
5
- import sklearn
6
-
7
- model= pickle.load(open('ALL_MODELS.pkl', 'rb'))
8
 
9
  def preprocess_image(img):
10
  import cv2
@@ -89,6 +86,7 @@ def get_feature(img):
89
  imdist = cv2.resize(imdist, (0, 0), fx=0.5, fy=0.5, interpolation=cv2.INTER_NEAREST)
90
  return feat
91
 
 
92
 
93
  def classify_image(image):
94
  X_test = get_feature(image)
 
2
  import argparse
3
  import numpy as np
4
  import pickle
 
 
 
5
 
6
  def preprocess_image(img):
7
  import cv2
 
86
  imdist = cv2.resize(imdist, (0, 0), fx=0.5, fy=0.5, interpolation=cv2.INTER_NEAREST)
87
  return feat
88
 
89
+ model= pickle.load(open('ALL_MODELS.pkl', 'rb'))
90
 
91
  def classify_image(image):
92
  X_test = get_feature(image)