some font changes

app.py

@@ -6,7 +6,7 @@ from torchvision.transforms import v2 as transforms
 import os
 
 # Define the class names
-class_names = ['AI-Generated Image', "Real/Non-AI-Generated Image"]
+class_names = ['Fake/AI-Generated Image', "Real/Not an AI-Generated Image"]
 
 # Load the model
 weights_path = "FaKe-ViT-B16.pth"
@@ -24,7 +24,8 @@ preprocess = transforms.Compose([
 def predict_image(image):
     image = preprocess(image)
     if image.shape[0] != 3:
-        image = image[:3, :, :]
+        # image = image[:3, :, :]
+        return "Invalid Image: Image should be in RGB format. Please upload a valid image."
     image = image.unsqueeze(0)
     with torch.inference_mode():
         output = model(image)
@@ -40,12 +41,14 @@ demo = gr.Interface(
     flagging_options=["incorrect prediction"],
      examples=[
          ("images/cheetah.jpg"),
-         ( "images/horse.jpg"),
+         ( "images/cat.jpg"),
          ("images/astronaut.jpg"),
+         ("images/mountain.jpg"),
+         ("images/unicorn.jpg") 
      ],
-    title="FaKe-ViT-B/16: AI-Generated Image Detection using Vision Transformer(ViT-B/16)",
-    description="This is a demo to detect AI-Generated images using Vision Transformer(ViT-B/16). Upload an image and the model will predict whether the image is AI-Generated or Real",
-    article=" \nBased on the paper:'An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale', Alexey et al.\nDataset: 'Fake or Real competition dataset' at https://huggingface.co/datasets/mncai/Fake_or_Real_Competition_Dataset"
+    title="<u>FaKe-ViT-B/16: Robust and Fast AI-Generated Image Detection using Vision Transformer(ViT-B/16):</u>",
+    description="<p style='font-size: 20px;'>This is a demo to detect AI-Generated images using a fine-tuned Vision Transformer(ViT-B/16). Upload an image and the model will predict whether the image is AI-Generated or Real",
+    article="<p style='font-size: 20px;'><b>Paper</b>: 'An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale', Alexey et al.<br/><b>Dataset</b>: 'Fake or Real competition dataset' at <a href='https://huggingface.co/datasets/mncai/Fake_or_Real_Competition_Dataset'>Fake or Real competition dataset</a>"
 )
 
 if __name__ == "__main__":