Update Traffic_Signs_Classification.py

Traffic_Signs_Classification.py

@@ -9,7 +9,7 @@ import torch
 from IPython.display import Audio
 
 # Streamlit application title
-st.title("Speech the Traffic Signs")
+st.title("Phonically describe traffic signs")
 
 #Traffic Sign Classification
 model= ViTForImageClassification.from_pretrained('Rae1230/Traffic_Signs_Classification')
@@ -33,7 +33,8 @@ if uploaded_file is not None:
     text_col = df['Name']
     
     text_value = text_col.loc[num_col == img_class_idx].values[0]
-    st.write("Predicted class:", text_value)
+    st.subtitle("Predicted traffic sign:")
+    st.write(text_value)
 
 
 #speech the Traffic Sign
@@ -46,5 +47,6 @@ if uploaded_file is not None:
     
     with torch.no_grad():
         output = model2(**inputs).waveform
-    
+        
+    st.subtitle("Phonically describe this traffic sign:")
     st.audio(output.numpy(),sample_rate=model2.config.sampling_rate)