Update app.py

app.py

@@ -1,4 +1,4 @@
-rom transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 def join_keywords(keywords, randomize=True):
   N = len(keywords)
@@ -19,7 +19,7 @@ device = torch.device("cuda")
 tokenizer = AutoTokenizer.from_pretrained("NihalSrivastava/advertisement-description-generator")
 model = AutoModelForCausalLM.from_pretrained("NihalSrivastava/advertisement-description-generator").to(device)
 
-keywords = ['laptop', 'fast', 'gaming', 'great graphics', 'affordable']
+keywords = ['mobile', 'application', 'children', 'image sharing', 'security']
 kw = join_keywords(keywords, randomize=False)
 prompt = SPECIAL_TOKENS['bos_token'] + kw + SPECIAL_TOKENS['sep_token']
 generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)