Spaces:
Paused
Paused
| from flask import Flask | |
| import ollama | |
| from util import time_model | |
| app = Flask(__name__) | |
| def hello_world(): | |
| return "<h1>Hello, World!<h1>" | |
| def test_world(): | |
| from ollama import ListResponse, list as ollist | |
| response: ListResponse = ollist() | |
| ansver = [] | |
| for model in response.models: | |
| ansver.append(f'<h4>Name:{model.model}</h4>') | |
| ansver.append(f'\t Size (MB): {(model.size.real / 1024 / 1024):.2f}') | |
| if model.details: | |
| ansver.append(f'\t Format: {model.details.format}') | |
| ansver.append(f'\t Family: {model.details.family}') | |
| ansver.append(f'\t Parameter Size: {model.details.parameter_size}') | |
| ansver.append(f'\t Quantization Level:{model.details.quantization_level}') | |
| result = [f'<p>{answ}</p>' for answ in ansver ] | |
| return '\n'.join(result) | |
| def ps(): | |
| from ollama import ProcessResponse, chat, ps, pull | |
| ansver = [] | |
| response: ProcessResponse = ps() | |
| for model in response.models: | |
| ansver.append(f'<h4>Model</h4>: {model.model}') | |
| ansver.append(f' \t Digest: {model.digest}') | |
| ansver.append(f' \t Expires at: {model.expires_at}') | |
| ansver.append(f' \t Size: {model.size}') | |
| ansver.append(f' \t Size vram: {model.size_vram}') | |
| ansver.append(f' \t Details: {model.details}') | |
| ansver.append(f' \t Context length: {model.context_length}') | |
| ansver.append(f'\n') | |
| result = [f'<p>{answ}</p>' for answ in ansver ] | |
| return '\n'.join(result) | |
| def test_time(): | |
| return time_model('qwen3:0.6b') | |
| def test_time17(): | |
| return time_model('qwen3:1.7b') | |
| def test_time4(): | |
| return time_model('qwen3:4b') | |
| def test_time8(): | |
| return time_model('qwen3:8b') | |
| def test_time14(): | |
| return time_model('qwen3:14b') | |
| #@app.route('/time30') | |
| def test_time30(): | |
| return time_model('qwen3:30b') | |
| if __name__ == '__main__': | |
| app.run(host='0.0.0.0', port="7860") | |