reduce inference size

app.py

@@ -8,7 +8,7 @@ from typing import Tuple, Dict
 def predict(img) -> Tuple[Dict, float]:
     start_time = timer()
     classifier = pipeline("image-classification", model="bazyl/gtsrb-model")
-    result = classifier(img)
+    result = classifier(img, top_k=3)
     response = {result[i]["label"]: result[i]["score"] for i in range(len(result))}
     pred_time = round(timer() - start_time, 5)
     return response, pred_time
@@ -24,7 +24,7 @@ demo = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),
     outputs=[
-        gr.Label(num_top_classes=5, label="Predictions"),
+        gr.Label(num_top_classes=3, label="Predictions"),
         gr.Number(label="Prediction time (s)"),
     ],
     examples=example_list,