muhammadfiaz commited on
Commit
1a0a30c
1 Parent(s): db174a0

Update webui.py

Browse files
Files changed (1) hide show
  1. webui.py +2 -2
webui.py CHANGED
@@ -18,8 +18,8 @@ with gr.Blocks(theme=gr.themes.Soft(), title="GemGPT") as app:#head="<link rel='
18
  gr.HTML("<h3 style='text-align: center;'>Talk to GemGPT, Powered by Gemma Models</h3>")
19
  with gr.Row():
20
  gr.Markdown("Select a model to run. Gemma-2b-it is a smaller model that is faster and uses less memory. Gemma-7b-it is a larger model that is slower and uses more memory.")
21
- model_options = gr.Dropdown(label="Select a Model", choices=["google/gemma-2b-it", "google/gemma-7b-it"],
22
- value="google/gemma-2b-it")
23
  with gr.Row():
24
  gr.Markdown("Select the device to run the model on. If you are running this on a CPU, select CPU. If you are running this on a GPU, select CUDA.")
25
  device = gr.Dropdown(label="Device", choices=["cuda", "cpu"], value=device)
 
18
  gr.HTML("<h3 style='text-align: center;'>Talk to GemGPT, Powered by Gemma Models</h3>")
19
  with gr.Row():
20
  gr.Markdown("Select a model to run. Gemma-2b-it is a smaller model that is faster and uses less memory. Gemma-7b-it is a larger model that is slower and uses more memory.")
21
+ model_options = gr.Dropdown(label="Select a Model", choices=["muhammadfiaz/gemma-2b-it", "google/gemma-7b-it"],
22
+ value="muhammadfiaz/gemma-2b-it")
23
  with gr.Row():
24
  gr.Markdown("Select the device to run the model on. If you are running this on a CPU, select CPU. If you are running this on a GPU, select CUDA.")
25
  device = gr.Dropdown(label="Device", choices=["cuda", "cpu"], value=device)