Update Traffic_Signs_Classification.py

Traffic_Signs_Classification.py

@@ -33,7 +33,7 @@ if uploaded_file is not None:
     text_col = df['Name']
     
     text_value = text_col.loc[num_col == img_class_idx].values[0]
-    st.head("Predicted traffic sign:")
+    st.header("Predicted traffic sign:")
     st.write(text_value)
 
 
@@ -48,5 +48,5 @@ if uploaded_file is not None:
     with torch.no_grad():
         output = model2(**inputs).waveform
         
-    st.head("Phonically describe this traffic sign:")
+    st.header("Phonically describe this traffic sign:")
     st.audio(output.numpy(),sample_rate=model2.config.sampling_rate)