Selma commited on
Commit
9aaffac
1 Parent(s): 5b07aff

adding docstring to the function

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -24,6 +24,15 @@ with open('labels.txt', "r") as f:
24
 
25
 
26
  def classify(image_raw):
 
 
 
 
 
 
 
 
 
27
 
28
  ## preprocessing
29
  # we need a transform step to normalise the pictures
24
 
25
 
26
  def classify(image_raw):
27
+ """
28
+ Takes an image in PIL format as input and returns a text that gives the most likely label to the image, along with its score according to resnet34
29
+
30
+ :param image_raw: The image to be classified/labeled
31
+ :type image_raw: PIL image
32
+
33
+ :returns: a list of strings representing the header columns
34
+ :rtype: str
35
+ """
36
 
37
  ## preprocessing
38
  # we need a transform step to normalise the pictures