Update app.py
Browse filesFix one last variable scope oversight...
app.py
CHANGED
@@ -324,7 +324,7 @@ class Applicant:
|
|
324 |
self.resume = resume
|
325 |
|
326 |
@spaces.GPU
|
327 |
-
def write(inputs, max_new_tokens
|
328 |
_output = model.generate(**inputs, max_new_tokens=max_new_tokens, do_sample=True, temperature=0.6, top_k=40, top_p=0.9, repetition_penalty=1.1)
|
329 |
return _output
|
330 |
|
|
|
324 |
self.resume = resume
|
325 |
|
326 |
@spaces.GPU
|
327 |
+
def write(inputs, max_new_tokens, do_sample=True, temperature=0.6, top_k=40, top_p=0.9, repetition_penalty=1.1):
|
328 |
_output = model.generate(**inputs, max_new_tokens=max_new_tokens, do_sample=True, temperature=0.6, top_k=40, top_p=0.9, repetition_penalty=1.1)
|
329 |
return _output
|
330 |
|