RoniFinTech commited on
Commit
84f2663
1 Parent(s): 5d67adc

update version

Browse files
Files changed (1) hide show
  1. main.py +5 -4
main.py CHANGED
@@ -28,10 +28,11 @@ refiner.enable_attention_slicing()
28
  app = FastAPI()
29
 
30
 
31
- # Define a function to handle the GET request at `/generate`
32
- # The generate() function is defined as a FastAPI route that takes a
33
- # string parameter called text. The function generates text based on the # input using the pipeline() object, and returns a JSON response
34
- # containing the generated text under the key "output"
 
35
  @app.get("/generate")
36
  def generate(text: str):
37
  """
 
28
  app = FastAPI()
29
 
30
 
31
+ @app.get("/")
32
+ async def root():
33
+ return {"message": "UNIK ML API"}
34
+
35
+
36
  @app.get("/generate")
37
  def generate(text: str):
38
  """