Spaces:
Build error
Build error
Abdelmageed
commited on
Commit
•
7e6e0da
1
Parent(s):
918cd3a
edit
Browse files
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 |
-
|
|
|
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 |
|