Update main.py

main.py

@@ -1,17 +1,17 @@
-from fastapi import FastAPI
-from app import predict
-import os
-from huggingface_hub import login
-
-os.environ['HF_HOME'] = '/hug/cache/'
-os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
-
-app = FastAPI()
-
-@app.get("/")
-async def root():
- return {"Code Review Automation":"Version 1.0 'First Draft'"}
-
-@app.post("/AutomateReview/")
-def predict(input_json: str):
+from fastapi import FastAPI
+from app import predict
+import os
+from huggingface_hub import login
+
+os.environ['HF_HOME'] = '/hug/cache/'
+os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
+
+app = FastAPI()
+
+@app.get("/")
+def root():
+ return {"Code Review Automation":"Version 1.0 'First Draft'"}
+
+@app.post("/AutomateReview/")
+def predict(input_json: str):
     return predict(input_json)