Spaces:
Sleeping
Sleeping
Update webui.py
Browse files
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=["
|
22 |
-
value="
|
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)
|