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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def image_to_query(imagefile):
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
 
 
33
  output: Query for the LLM
34
  """
35
  #image = Image.open(imagefile)
36
+ classifier = pipeline("image-classification", model = "nprasad24/bean_classifier", from_tf = True)
37
 
38
  scores = classifier(imagefile)
39