Diamanta commited on
Commit
72505c7
·
verified ·
1 Parent(s): 9d9d39a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py CHANGED
@@ -48,6 +48,24 @@ def load_model():
48
  async def health_check():
49
  return {"status": "ok"}
50
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  @app.get("/models")
52
  async def list_models():
53
  # Return available models info
 
48
  async def health_check():
49
  return {"status": "ok"}
50
 
51
+ @app.get("/api/tags")
52
+ async def api_tags():
53
+ return {
54
+ "models": [
55
+ {
56
+ "name": "llama2",
57
+ "modified_at": "2024-06-01T00:00:00Z",
58
+ "size": 4200000000,
59
+ "digest": "sha256:placeholder",
60
+ "details": {
61
+ "format": "gguf",
62
+ "family": "llama",
63
+ "families": ["llama"]
64
+ }
65
+ }
66
+ ]
67
+ }
68
+
69
  @app.get("/models")
70
  async def list_models():
71
  # Return available models info