Update app.py

app.py

@@ -4,13 +4,17 @@ import torch
 from PIL import Image
 from io import BytesIO
 
-# Load the fine-tuned model
-model_path = '/content/model_after_5_epochs.pth'
-model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
-model.load_state_dict(torch.load(model_path))
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model.to(device)
-tokenizer = AutoTokenizer.from_pretrained("Salesforce/blip-image-captioning-base")
+# # Load the fine-tuned model
+# model_path = '/content/model_after_5_epochs.pth'
+# model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
+# model.load_state_dict(torch.load(model_path))
+# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# model.to(device)
+# tokenizer = AutoTokenizer.from_pretrained("Salesforce/blip-image-captioning-base")
+
+
+model = BlipForConditionalGeneration.from_pretrained("MLInAi/CartoonCaptionGen")
+tokenizer = AutoTokenizer.from_pretrained("MLInAi/CartoonCaptionGen")
 
 # Function to generate caption for the uploaded image
 def generate_caption(image):