Spaces:

Abdelmageed95
/

Social_media_post_generation

Build error

Abdelmageed commited on Sep 22, 2022

Commit

7c97898

•

1 Parent(s): 4ec868f

dd

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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")
   res = generate_post( model_bm , tokenizer_bm, company_name , description , example1 , example2 , example3)
   generated_caption = generate_caption( model_cp ,

   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-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 ,