Update TextGen/router.py

TextGen/router.py

@@ -6,7 +6,7 @@ from fastapi import FastAPI, HTTPException, Query, Request
 from fastapi.responses import StreamingResponse,Response
 from fastapi.middleware.cors import CORSMiddleware
 
-from mistralai import Mistral
+from mistralai.client import Mistral
 from TextGen.suno import custom_generate_audio, get_audio_information,generate_lyrics
 from TextGen.gemini import generate_story,place_objects,generate_map_markdown
 #from TextGen.diffusion import generate_image