Ankitajadhav commited on
Commit
699b14f
·
verified ·
1 Parent(s): 55952b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -71,9 +71,9 @@ def generate_text(
71
  temp += out["choices"][0]["text"]
72
  yield temp
73
 
74
- # Define the Gradio interface without timeout
75
  demo = gr.Interface(
76
- fn=generate_text,
77
  inputs="text",
78
  outputs="text",
79
  title="llama-cpp-python on GPU with ChromaDB",
@@ -91,7 +91,6 @@ demo = gr.Interface(
91
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
92
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
93
  ],
94
- processing_timeout=None, # No timeout limit
95
  )
96
 
97
  if __name__ == "__main__":
 
71
  temp += out["choices"][0]["text"]
72
  yield temp
73
 
74
+ # Define the Gradio interface without retry_btn
75
  demo = gr.Interface(
76
+ fn=generate_text,
77
  inputs="text",
78
  outputs="text",
79
  title="llama-cpp-python on GPU with ChromaDB",
 
91
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
92
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
93
  ],
 
94
  )
95
 
96
  if __name__ == "__main__":