Spaces:

UcsTurkey
/

fine-tune-inference-test

Paused

ciyidogan commited on May 13

Commit

b5c1d21

verified ·

1 Parent(s): 584aa58

Update fine_tune_inference_test.py

Files changed (1) hide show

fine_tune_inference_test.py CHANGED Viewed

@@ -34,7 +34,11 @@ pipe = None  # global text-generation pipeline
 class Message(BaseModel):
     user_input: str
-@app.get("/", response_class=HTMLResponse)
 def root():
     return """
     <html>

 class Message(BaseModel):
     user_input: str
+@app.get("/")
+def health():
+    return JSONResponse(content={"status": "ok"})
+@app.get("/start", response_class=HTMLResponse)
 def root():
     return """
     <html>