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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,7 +24,7 @@ transformers.logging.set_verbosity_error()
24
 
25
 
26
 
27
- def image_to_query(image):
28
  """
29
  input: Image
30
 
@@ -32,7 +32,7 @@ def image_to_query(image):
32
 
33
  output: Query for the LLM
34
  """
35
- image = Image.open('test.jpg')
36
  classifier = pipeline("image-classification", model = "nprasad24/bean_classifier")
37
 
38
  scores = classifier(image)
 
24
 
25
 
26
 
27
+ def image_to_query(imagefile):
28
  """
29
  input: Image
30
 
 
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)