Update Traffic_Signs_Classification.py

Traffic_Signs_Classification.py

@@ -18,11 +18,19 @@ st.title("Speech the Traffic Signs")
 uploaded_file = st.file_uploader("Choose a PNG image...", type="png", accept_multiple_files=False)
 if uploaded_file is not None:
     img = Image.open(uploaded_file)
+    st.image(img, caption='Uploaded Image.', use_column_width=True)
     inputs = processor(img.convert('RGB'), return_tensors="pt")
 
     outputs = model(**inputs)
 
     logits = outputs.logits
     img_class_idx=logits.argmax(-1).item()
-
-    st.write(img_class_idx)
+    with open("/TrafficSigns_Classification/labels.csv", "r") as file:
+        df = pd.read_csv(file)
+    
+    
+    num_col = df['ClassId']
+    text_col = df['Name']
+    
+    text_value = text_col.loc[num_col == img_class_idx].values[0]
+    st.write("Predicted class:", text_value)