@@ -319,27 +319,35 @@ def create_interface():
319
319
Choose your preferred model for the conversation.
320
320
""" )
321
321
322
- model_dropdown = gr .Dropdown (
323
- choices = model_choices ,
324
- value = default_model ,
325
- label = "Select Model" ,
326
- info = "Choose the model to use for the conversation"
327
- )
322
+ with gr .Row ():
323
+ with gr .Column ():
324
+ model_dropdown = gr .Dropdown (
325
+ choices = model_choices ,
326
+ value = default_model ,
327
+ label = "Select Model" ,
328
+ info = "Choose the model to use for the conversation"
329
+ )
330
+ download_button = gr .Button ("Download Selected Model" )
331
+ model_status = gr .Textbox (
332
+ label = "Download Status" ,
333
+ placeholder = "Select a model and click Download to begin..." ,
334
+ interactive = False
335
+ )
328
336
329
337
# Add model FAQ section
330
338
gr .Markdown ("""
331
339
## Model FAQ
332
340
333
341
| Model | Parameters | Size | Download Command |
334
342
|-------|------------|------|------------------|
335
- | qwq | 7B | 4.1GB | qwq:latest |
336
- | gemma3 | 7B | 4.1GB | gemma3:latest |
337
- | llama3.3 | 7B | 4.1GB | llama3.3:latest |
338
- | phi4 | 7B | 4 .1GB | phi4:latest |
343
+ | qwq | 32B | 20GB | qwq:latest |
344
+ | gemma3 | 4B | 3.3GB | gemma3:latest |
345
+ | llama3.3 | 70B | 43GB | llama3.3:latest |
346
+ | phi4 | 14B | 9 .1GB | phi4:latest |
339
347
| mistral | 7B | 4.1GB | mistral:latest |
340
- | llava | 7B | 4.1GB | llava:latest |
341
- | phi3 | 7B | 4.1GB | phi3:latest |
342
- | deepseek-r1 | 7B | 4.1GB | deepseek-r1:latest |
348
+ | llava | 7B | 4.5GB | llava:latest |
349
+ | phi3 | 4B | 4.0GB | phi3:latest |
350
+ | deepseek-r1 | 7B | 4.7GB | deepseek-r1:latest |
343
351
344
352
Note: All models are available through Ollama. Make sure Ollama is running on your system.
345
353
""" )
@@ -434,6 +442,9 @@ def create_interface():
434
442
url_button .click (process_url , inputs = [url_input ], outputs = [url_output ])
435
443
repo_button .click (process_repo , inputs = [repo_input ], outputs = [repo_output ])
436
444
445
+ # Model download event handler
446
+ download_button .click (download_model , inputs = [model_dropdown ], outputs = [model_status ])
447
+
437
448
# Standard chat handlers
438
449
standard_msg .submit (
439
450
chat ,
0 commit comments