Update app.py

app.py

@@ -2,8 +2,8 @@ import torch
 from transformers import AutoProcessor,AutoModelForCausalLM
 import gradio as gr 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-processor=AutoProcessor.from_pretrained("saravanamax/General_image_captioning")
-model=AutoModelForCausalLM.from_pretrained("saravanamax/General_image_captioning").to(device)
+processor=AutoProcessor.from_pretrained("saravanamax/image_captioning")
+model=AutoModelForCausalLM.from_pretrained("saravanamax/image_captioning").to(device)
 def generate_caption(image):
   encoded=processor(images=image, return_tensors="pt").to(device)
   pixels=encoded['pixel_values'].to(device)