merve HF staff commited on
Commit
b42a3ee
1 Parent(s): f9b32f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -96,9 +96,9 @@ def categorize(input):
96
  return ttc(input)[0]["generated_text"]
97
 
98
  @spaces.GPU
99
- def turna(input, max_new_tokens = max_new_tokens, length_penalty=length_penalty,
100
- top_k=top_k, top_p=top_p, temperature=temp, num_beams=num_beams,
101
- do_sample = do_sample):
102
  turna = pipeline(model="boun-tabi-LMG/TURNA", device=0)
103
 
104
  return turna(input, max_new_tokens = max_new_tokens, length_penalty=length_penalty,
 
96
  return ttc(input)[0]["generated_text"]
97
 
98
  @spaces.GPU
99
+ def turna(input, max_new_tokens, length_penalty,
100
+ top_k, top_p, temp, num_beams,
101
+ do_sample):
102
  turna = pipeline(model="boun-tabi-LMG/TURNA", device=0)
103
 
104
  return turna(input, max_new_tokens = max_new_tokens, length_penalty=length_penalty,