dd

app.py

@@ -57,8 +57,8 @@ def demo_smg(company_name ,description , example1 , example2 , example3):
   model_cp= T5ForConditionalGeneration.from_pretrained("Abdelmageed95/caption_model" , use_auth_token = access_token )
   tokenizer = T5Tokenizer.from_pretrained('t5-base')
 
-  model_bm = AutoModelForCausalLM.from_pretrained("bigscience/bloom-3b" , use_auth_token = access_token)
-  tokenizer_bm = AutoTokenizer.from_pretrained("bigscience/bloom-3b")
+  model_bm = AutoModelForCausalLM.from_pretrained("bigscience/bloom-560m" , use_auth_token = access_token)
+  tokenizer_bm = AutoTokenizer.from_pretrained("bigscience/bloom-560m")
   
   res = generate_post( model_bm , tokenizer_bm, company_name , description , example1 , example2 , example3)
   generated_caption = generate_caption( model_cp ,