Update app.py

app.py

@@ -3,9 +3,9 @@ import tensorflow as tf
 import gradio as gr
 from huggingface_hub import from_pretrained_keras
 
-teacher_model = from_pretrained_keras("Blazer007/consistency_training_with_supervision_teacher_model")
+teacher_model = from_pretrained_keras("keras-io/consistency_training_with_supervision_teacher_model")
 
-student_model = from_pretrained_keras("Blazer007/consistency_training_with_supervision_student_model")
+student_model = from_pretrained_keras("keras-io/consistency_training_with_supervision_student_model")
 
 class_names = [
     "Airplane",
@@ -54,10 +54,10 @@ input = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE))
 output = [gr.outputs.Label(label = "Teacher Model Output"), gr.outputs.Label(label = "Student Model Output")]
 
 title = "Image Classification using Consistency training with supervision"
-description = "Upload an image or select from examples to classify it.<br>The allowed classes are - Airplane, Automobile, Bird, Cat, Deer, Dog, Frog, Horse, Ship, Truck.<br><p><b>Teacher Model Repo - </b> <br><b> Student Model Repo -  </b></p>"
+description = "Upload an image or select from examples to classify it.<br>The allowed classes are - Airplane, Automobile, Bird, Cat, Deer, Dog, Frog, Horse, Ship, Truck.<br><p><b>Teacher Model Repo - https://huggingface.co/keras-io/consistency_training_with_supervision_teacher_model</b> <br><b> Student Model Repo - https://huggingface.co/keras-io/consistency_training_with_supervision_student_model </b><br><b>Keras Example - https://keras.io/examples/vision/consistency_training/</b></p>"
 
 
-article = "<div style='text-align: center;'><a href='https://twitter.com/_Blazer_007' target='_blank'>Space by Vivek Rai</a><br><a href='https://keras.io/examples/vision/consistency_training/' target='_blank'>Keras example by Sayak Paul</a></div>"
+article = "<div style='text-align: center;'><a href='https://twitter.com/_Blazer_007' target='_blank'>Space by Vivek Rai</a><br><a href='https://twitter.com/RisingSayak' target='_blank'>Keras example by Sayak Paul</a></div>"
 
 gr_interface = gr.Interface(
     infer,