mrudulkolambe commited on
Commit
e7cc263
1 Parent(s): d21bca2

Update app.py

Browse files

removed caption

Files changed (1) hide show
  1. app.py +0 -12
app.py CHANGED
@@ -45,18 +45,6 @@ def init_model(init_model_required):
45
  return processor, model, init_model_required
46
 
47
  #def main():
48
-
49
- st.header("Automate Fashion Image Captioning using BLIP-2")
50
- st.caption("The fashion industry is worth trillions of dollars. The goal of any company/seller is to help customer tofind the right product from a huge corpus of products that they are searching for.")
51
- st.caption("So, when customer find the right product they are mostly going to add the item to their cart and which help in company revenue.")
52
- st.caption("Accurate and enchanting descriptions of clothes on shopping websites can help customers without fashion knowledge to better understand the features (attributes, style, functionality, etc.) of the items and increase online sales by enticing more customers.")
53
- st.caption("Also, most of the time when any customer visits shopping websites, they are looking for a certain style or type of clothes that wish to purchase, they search for the item by providing a description of the item and the system finds the relevant items that match the search query by computing the similarity score between the query and the item caption.")
54
- st.caption("Given the clothes image provide a short caption that describes the item. In general, in image captioning datasets (e.g., COCO, Fliker), the descriptions of fashion items have three unique features, which makes the automatic generation of captions a challenging task. First, fashion captioning needs to describe the attributes of an item, while image captioning generally narrates the objects and their relations in the image.")
55
- st.caption("Solution: Used Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Models [(BLIP-2)](https://huggingface.co/Salesforce/blip2-opt-2.7b) by Salesforce. The original model size was too large. It was quite challenging to fit and fine-tune the model on the 16GB GPU.")
56
- st.caption("So, for this project have downloaded the pre-trained model [ybelkada/blip2-opt-2.7b-fp16-sharded](https://huggingface.co/ybelkada/blip2-opt-2.7b-fp16-sharded). This model uses OPT-2.7b LLM model with reduced precision to float16.")
57
- #st.caption("")
58
- st.caption("For more detail check out [github](https://github.com/SmithaUpadhyaya/fashion_image_caption)") #write
59
-
60
  #Select few sample images for the catagory of cloths
61
  with st.form("app", clear_on_submit = True):
62
 
 
45
  return processor, model, init_model_required
46
 
47
  #def main():
 
 
 
 
 
 
 
 
 
 
 
 
48
  #Select few sample images for the catagory of cloths
49
  with st.form("app", clear_on_submit = True):
50