Hotfix

Dockerfile

@@ -1,16 +1,21 @@
 # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.9
+FROM python:3.12
 
 RUN useradd -m -u 1000 user
-USER user
+USER root
 ENV PATH="/home/user/.local/bin:$PATH"
 
-WORKDIR /app
+WORKDIR /
+
+RUN apt-get update && apt-get install -y \
+    libgl1 \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
 
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,5 +1,4 @@
 from fastapi import FastAPI
-from transformers import AutoModelForCausalLM, AutoTokenizer
 from pydantic import BaseModel
 from langchain_ollama import OllamaLLM
 import uvicorn
@@ -18,7 +17,7 @@ def greet_json():
 
 @app.post("/generate_chat")
 def generateAi(request: ChatRequest):
-    result = llmModel.invoke(input="Hello, how are you?")
+    result = llmModel.invoke(input=request.prompt)
     
     return {"answer": result}