kek

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 #sentencepiece no installed??
 
 translator = pipeline("translation_en_to_ru", model="Helsinki-NLP/opus-mt-en-ru")
-image_classifier = pipeline("image-classification", model="microsoft/resnet-50", use_fast=False)
+image_classifier = pipeline("image-classification", model="microsoft/resnet-50")
 
 def translate_to_russian(text):
     try:
@@ -17,8 +17,12 @@ 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
     return output
     
 with gr.Blocks() as demo: