Update app.py

app.py

@@ -5,7 +5,7 @@ import numpy as np
 from PIL import Image
 import io
 
-# Load pre-trained deblurring model
+# Load the model from Hugging Face Hub using from_pretrained_keras
 model = from_pretrained_keras("google/maxim-s3-deblurring-reds")
 
 def deblur_image(input_image):
@@ -14,10 +14,10 @@ def deblur_image(input_image):
     image = tf.convert_to_tensor(image)
     image = tf.image.resize(image, (256, 256))
     
-    # Predict with the model
+    # Make predictions with the model
     predictions = model.predict(tf.expand_dims(image, 0))
     
-    # Convert back to image
+    # Convert the prediction back to an image
     output_image = predictions[0].numpy().astype(np.uint8)
     output_image = Image.fromarray(output_image)
     
@@ -35,4 +35,4 @@ iface = gr.Interface(fn=deblur_image,
                      live=True)
 
 # Launch the Gradio interface
-iface.launch()
+iface.launch()