Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|
17 |
-
prompt = f'
|
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"
|
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()
|