Abdelmageed commited on
Commit
7e6e0da
1 Parent(s): 918cd3a
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,10 +53,10 @@ def generate_caption(model , text_body ,tokenizer ,max_length):
53
 
54
  def demo_smg(company_name ,description , example1 , example2 , example3):
55
 
56
- model_cp= T5ForConditionalGeneration.from_pretrained("Abdelmageed95/caption_model")
 
57
  tokenizer = T5Tokenizer.from_pretrained('t5-base')
58
 
59
- access_token = "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"
60
  model_bm = AutoModelForCausalLM.from_pretrained("bigscience/bloom-3b" , use_auth_token = access_token)
61
  tokenizer_bm = AutoTokenizer.from_pretrained("bigscience/bloom-3b")
62
 
 
53
 
54
  def demo_smg(company_name ,description , example1 , example2 , example3):
55
 
56
+ access_token = "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"
57
+ model_cp= T5ForConditionalGeneration.from_pretrained("Abdelmageed95/caption_model" , use_auth_token = access_token )
58
  tokenizer = T5Tokenizer.from_pretrained('t5-base')
59
 
 
60
  model_bm = AutoModelForCausalLM.from_pretrained("bigscience/bloom-3b" , use_auth_token = access_token)
61
  tokenizer_bm = AutoTokenizer.from_pretrained("bigscience/bloom-3b")
62