changes

Dockerfile

@@ -21,4 +21,4 @@ ENV HOME=/home/user \
 
 COPY --chown=user . $HOME/app
 
-ENTRYPOINT ./entrypoint.sh
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -7,6 +7,11 @@ classifier = pipeline("zero-shot-classification", model='cross-encoder/nli-deber
 
 
 @app.get("/")
+async def root(title):
+    return {"Hello": "World!"}
+
+
+@app.get("/predict")
 async def root(title):
     categories = ["technology", "sports", "politics", "weather", "business", "entertainment"]
     response = classifier(title, categories)