Spaces:

Nishthaaa
/

my_space

Sleeping

Nishthaaa commited on Mar 18

Commit

bbe5e0c

1 Parent(s): bad6934

model name added

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 # Load model and processor
 @st.cache_resource
 def load_model():
-    model_name = "your-huggingface-username/your-model-name"
     processor = BlipProcessor.from_pretrained(model_name)
     model = BlipForConditionalGeneration.from_pretrained(model_name)
     return processor, model

 # Load model and processor
 @st.cache_resource
 def load_model():
+    model_name = "Nishthaaa/image_captioning"
     processor = BlipProcessor.from_pretrained(model_name)
     model = BlipForConditionalGeneration.from_pretrained(model_name)
     return processor, model