0x7o
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ if torch.cuda.is_available():
|
|
10 |
|
11 |
@spaces.GPU
|
12 |
def predict(prompt, temperature, max_length):
|
13 |
-
return pipe(prompt, temperature=temperature, max_length=max_length)[0]["generated_text"]
|
14 |
|
15 |
demo = gr.Interface(
|
16 |
fn=predict,
|
|
|
10 |
|
11 |
@spaces.GPU
|
12 |
def predict(prompt, temperature, max_length):
|
13 |
+
return pipe(prompt, temperature=temperature, max_length=max_length, top_p=0.95, top_k=50, do_sample=True)[0]["generated_text"]
|
14 |
|
15 |
demo = gr.Interface(
|
16 |
fn=predict,
|