nprasad24 commited on
Commit
f5fa604
·
verified ·
1 Parent(s): 0e4d269

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,10 +32,10 @@ def image_to_query(imagefile):
32
 
33
  output: Query for the LLM
34
  """
35
- image = Image.open(imagefile)
36
  classifier = pipeline("image-classification", model = "nprasad24/bean_classifier")
37
 
38
- scores = classifier(image)
39
 
40
  # Get the dictionary with the maximum score
41
  max_score_dict = max(scores, key=lambda x: x['score'])
 
32
 
33
  output: Query for the LLM
34
  """
35
+ #image = Image.open(imagefile)
36
  classifier = pipeline("image-classification", model = "nprasad24/bean_classifier")
37
 
38
+ scores = classifier(imagefile)
39
 
40
  # Get the dictionary with the maximum score
41
  max_score_dict = max(scores, key=lambda x: x['score'])