Update app.py

app.py

@@ -69,8 +69,18 @@ async def generate(request: Request):
 @app.get("/api/v1/models")
 async def get_models():
     try:
-        streamModel = TogetherAPI()
-        models = streamModel.get_model_list()
+        models = {
+            'text': {
+                'together': TogetherAPI().get_model_list(),
+                'xai': XaiAPI().get_model_list(),
+                'groq': GroqAPI().get_model_list(),
+                'deepinfra': DeepinfraAPI().get_model_list()
+            },
+            'image': {
+                'fal': FalAPI().get_model_list(),
+                'together': TogetherImageAPI().get_model_list()
+            }
+        }
         return {"models": models}
     except Exception as e:
         return {"error": f"An error occurred: {str(e)}"}