Update src/routes/ai_chat.py

src/routes/ai_chat.py

@@ -9,11 +9,11 @@ from flask_cors import cross_origin
 vgpu_path = os.path.join(os.path.dirname(__file__), '..', '..', '..', 'virtual_gpu_setup', 'virtual_gpu')
 sys.path.insert(0, vgpu_path)
 
-from virtual_gpu.vgpu.py import VirtualGPU
-from virtual_gpu.vram import VRAM
-from virtual_gpu.driver import GPUDriver
-from virtual_gpu.render import Renderer
-from virtual_gpu.ai import AIAccelerator
+from newm.virtual_gpu.vgpu.py import VirtualGPU
+from newm.virtual_gpu.vram import VRAM
+from newm.virtual_gpu.driver import GPUDriver
+from newm.virtual_gpu.render import Renderer
+from newm.virtual_gpu.ai import AIAccelerator
 import numpy as np
 
 # Import the Hugging Face GPT model from the same directory