Spaces:

tommy24
/

classifier

Running

tommy24 commited on Oct 18, 2023

Commit

4dbb024

1 Parent(s): 38a36a9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -75,7 +75,7 @@ def classify(image_path, text_input):
             reply = response["choices"][0]["message"]["content"]
-            output.append({"type": max_label, "prediction_value": round(max_prediction_value, 2), "content": reply})
         return output  # Return the populated output list
     else:

             reply = response["choices"][0]["message"]["content"]
+            output.append({"type": max_label, "prediction_value": rounded_value, "content": reply})
         return output  # Return the populated output list
     else: