Add endpoint

Dockerfile

@@ -8,8 +8,8 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-# 设置环境变量，指定 Hugging Face Transformers 缓存目录
-ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+# 设置环境变量，指定 Hugging Face 缓存目录
+ENV HF_HOME=/tmp/transformers_cache
 
 # 设置工作目录
 WORKDIR /app
@@ -27,3 +27,4 @@ EXPOSE 7860
 # 启动应用
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
+

app.py

@@ -34,6 +34,11 @@ gpt2_model = AutoModelForCausalLM.from_pretrained("TracyShen301/myFinnishChatbot
 AUDIO_DIR = "/tmp/audio_files"
 os.makedirs(AUDIO_DIR, exist_ok=True)
 
+@app.get("/")
+async def read_root():
+    return {"message": "Welcome to the Finnish Chatbot API!"}
+
+
 @app.post("/process_audio/")
 async def process_audio(file: UploadFile = File(...)):
     try: