Shreyas094 commited on
Commit
a82a747
·
verified ·
1 Parent(s): b24c9b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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": 4000, # Adjust as needed
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, step=1, 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
  ],
 
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
  ],