ameya-akkalkotkar commited on
Commit
28e5697
·
1 Parent(s): c0c80ac

Update app.py

Browse files

Changed app.py to update project folder and model file

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from peft import PeftModel, PeftConfig
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
- peft_model_id = f"FourthBrainGenAI/BLOOMZ-3b-marketmail-ai"
6
  config = PeftConfig.from_pretrained(peft_model_id)
7
  model = AutoModelForCausalLM.from_pretrained(
8
  config.base_model_name_or_path,
@@ -39,6 +39,6 @@ if __name__ == "__main__":
39
  gr.inputs.Textbox(lines=5, label="Product Description"),
40
  ],
41
  gr.outputs.Textbox(label="Ad"),
42
- title="MarketMail-AI",
43
- description="MarketMail-AI is a tool that generates marketing emails for products.",
44
  ).launch()
 
2
  from peft import PeftModel, PeftConfig
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
+ peft_model_id = f"ameya-akkalkotkar/BloomVideoGameTweetGen"
6
  config = PeftConfig.from_pretrained(peft_model_id)
7
  model = AutoModelForCausalLM.from_pretrained(
8
  config.base_model_name_or_path,
 
39
  gr.inputs.Textbox(lines=5, label="Product Description"),
40
  ],
41
  gr.outputs.Textbox(label="Ad"),
42
+ title="VideoGamePromoAI",
43
+ description="A tool that generates marketing promo emails for video game products.",
44
  ).launch()