q

app.py

@@ -17,12 +17,8 @@ def translate_to_russian(text):
 def classify_image(image):
     results = image_classifier(image)
     output = {}
-    count = 0
     for result in results:
-        output[translate_to_russian(result['label'])] = result['score']
-        count += 1
-        if (count > 5):
-            break
+        output[result['label']] = result['score']
     return output
     
 with gr.Blocks() as demo: