Start building the CUDA container image

app/main.py

@@ -1,27 +1,7 @@
-from typing import Union
-from mistralai.client import MistralClient
-from mistralai.models.chat_completion import ChatMessage
 from fastapi import FastAPI
-import streamlit as st
-import os
 
 app = FastAPI()
 
-api_key = "RjxyIWwyffo7JdSmqQFuIFVl8zUjoLzi"
-model = "mistral-large-latest"
-
-client = MistralClient(api_key=api_key)
-
-
 @app.get("/")
-def read_root():
-    for response in client.chat_stream(
-            model=model,
-            messages=[ChatMessage(role="user", content="What is the best French cheese?")]
-        ):
-            print(response.choices[0].delta.content or "")
-#     chat_response = client.chat(
-#     model=model,
-#     messages=[ChatMessage(role="user", content="What is the best French cheese?")]
-# )
-    return {"Hello": chat_response.choices[0].message.content}
+def root():
+    return {"Hello": "World"}