Baskar2005 commited on
Commit
53727cd
1 Parent(s): 3dff0ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,7 +9,7 @@ from transformers import pipeline
9
 
10
  class Image_classification:
11
  def __init__(self):
12
- self.model=""
13
 
14
 
15
  def unzip_image_data(self) -> str:
@@ -65,8 +65,8 @@ class Image_classification:
65
  Dict[str, float]: A dictionary of classification labels and their corresponding scores.
66
  """
67
  try:
68
- self.model=model
69
- classifier = pipeline("image-classification", model=self.model)
70
  result= classifier(image)
71
  return result
72
  except Exception as e:
 
9
 
10
  class Image_classification:
11
  def __init__(self):
12
+ pass
13
 
14
 
15
  def unzip_image_data(self) -> str:
 
65
  Dict[str, float]: A dictionary of classification labels and their corresponding scores.
66
  """
67
  try:
68
+
69
+ classifier = pipeline("image-classification", model=model)
70
  result= classifier(image)
71
  return result
72
  except Exception as e: