Ali-C137 commited on
Commit
840bcf5
1 Parent(s): caf451f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -13,15 +13,15 @@ tokenizer = AutoTokenizer.from_pretrained('EleutherAI/gpt-neo-1.3B')
13
 
14
  set_seed(424242)
15
 
16
- def generate(Name, Employer, Position, Organization, Hard_skills, Soft_skills, max_length=500, top_k=1, temperature=0.9, repetition_penalty = 2.0):
17
- prompt = f'im {Name} and i want to write a motivation letter to {Employer} about the position {Position} at {Organization} mentioning the hard skills {Hard_skills} and soft skills {Soft_skills} you have acquired'
18
  input_ids = tokenizer(prompt, return_tensors="pt").to(0)
19
  sample = model.generate(**input_ids, max_length=max_length, top_k=top_k, temperature=temperature, repetition_penalty = repetition_penalty)
20
  return tokenizer.decode(sample[0], truncate_before_pattern=[r"\n\n^#", "^'''", "\n\n\n"])
21
 
22
- title = "Motivation Letter Generator w/ GPT-Neo-1.3B"
23
  article = "Impress your employer"
24
 
25
- gr = gr.Interface(fn=generate, inputs=["text", "text", "text", "text", "text", "text"], outputs="text", title=title, article=article)
26
 
27
  gr.launch()
 
13
 
14
  set_seed(424242)
15
 
16
+ def generate(Name, Position, Organization, max_length=500, top_k=1, temperature=0.9, repetition_penalty = 2.0):
17
+ prompt = f"i'm {Name} and i want to write a motivation letter to an employer about the position of {Position} at {Organization} mentioning the hard skills and soft skills i have acquired"
18
  input_ids = tokenizer(prompt, return_tensors="pt").to(0)
19
  sample = model.generate(**input_ids, max_length=max_length, top_k=top_k, temperature=temperature, repetition_penalty = repetition_penalty)
20
  return tokenizer.decode(sample[0], truncate_before_pattern=[r"\n\n^#", "^'''", "\n\n\n"])
21
 
22
+ title = "Motivation Letter Generator"
23
  article = "Impress your employer"
24
 
25
+ gr = gr.Interface(fn = generate, inputs=["text", "text", "text"], outputs="text", title=title, article=article)
26
 
27
  gr.launch()