Spaces:

archaiveproject
/

ImageCaptioning

Sleeping

Prakhar Trivedi commited on Jul 21, 2025

Commit

e1a9146

1 Parent(s): 2713ac2

fixed broken repo id reference

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ EMBED_DIM = 256
 HIDDEN_DIM = 512
 MAX_SEQ_LENGTH = 25
 VOCAB_SIZE = 8492
-DEVICE = torch.device("cpu")
 transform_inference = transforms.Compose([
     transforms.Resize((384, 384)),
@@ -151,7 +151,7 @@ def load_model_and_vocab(repo_id):
     return model, vocab
-model, vocab = load_model_and_vocab("prakhartrivedi/ImageCaptioningSpace")
 print("Model and vocabulary loaded successfully.")
 def generate_caption_for_image(img):

 HIDDEN_DIM = 512
 MAX_SEQ_LENGTH = 25
 VOCAB_SIZE = 8492
+DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 transform_inference = transforms.Compose([
     transforms.Resize((384, 384)),
     return model, vocab
+model, vocab = load_model_and_vocab("prakhartrivedi/ImageCaptioningSCCCI")
 print("Model and vocabulary loaded successfully.")
 def generate_caption_for_image(img):