Add application file

Dockerfile

@@ -1,16 +1,10 @@
-# 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
-
-RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
+FROM python:3.10
 
 WORKDIR /app
 
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+COPY requirements.txt .
+RUN pip install -r requirements.txt
+
+COPY . .
 
-COPY --chown=user . /app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,7 +1,15 @@
 from fastapi import FastAPI
+from transformers import pipeline
 
 app = FastAPI()
 
+classifier = pipeline("sentiment-analysis")
+
 @app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+def home():
+    return {"message": "API is running"}
+
+@app.get("/predict")
+def predict(text: str):
+    result = classifier(text)
+    return result

requirements.txt

@@ -1,2 +1,4 @@
 fastapi
-uvicorn[standard]
+uvicorn
+transformers
+torch