Update app.py
Browse files
app.py
CHANGED
|
@@ -124,7 +124,7 @@ def generate_text(input_text, temperature=0.7, repetition_penalty=1.0, top_p=0.9
|
|
| 124 |
data = {
|
| 125 |
"inputs": input_text,
|
| 126 |
"parameters": {
|
| 127 |
-
"max_new_tokens":
|
| 128 |
"temperature": temperature,
|
| 129 |
"repetition_penalty": repetition_penalty,
|
| 130 |
"top_p": top_p
|
|
@@ -279,7 +279,7 @@ gr.Interface(
|
|
| 279 |
gr.File(label="Upload PDF"),
|
| 280 |
gr.Slider(minimum=1, maximum=20, label="Number of Results"), # Added Slider for num_results
|
| 281 |
gr.Textbox(label="Instructions"),
|
| 282 |
-
gr.Slider(minimum=0.1, maximum=1.0,
|
| 283 |
gr.Slider(minimum=0.1, maximum=1.0, label="Repetition Penalty"),
|
| 284 |
gr.Slider(minimum=0.1, maximum=1.0, label="Top p")
|
| 285 |
],
|
|
|
|
| 124 |
data = {
|
| 125 |
"inputs": input_text,
|
| 126 |
"parameters": {
|
| 127 |
+
"max_new_tokens": 8000, # Adjust as needed
|
| 128 |
"temperature": temperature,
|
| 129 |
"repetition_penalty": repetition_penalty,
|
| 130 |
"top_p": top_p
|
|
|
|
| 279 |
gr.File(label="Upload PDF"),
|
| 280 |
gr.Slider(minimum=1, maximum=20, label="Number of Results"), # Added Slider for num_results
|
| 281 |
gr.Textbox(label="Instructions"),
|
| 282 |
+
gr.Slider(minimum=0.1, maximum=1.0, label="Temperature"),
|
| 283 |
gr.Slider(minimum=0.1, maximum=1.0, label="Repetition Penalty"),
|
| 284 |
gr.Slider(minimum=0.1, maximum=1.0, label="Top p")
|
| 285 |
],
|