Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 from fastapi import FastAPI, Query
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
@@ -16,7 +14,7 @@ def load_prompter():
 
 prompter_model, prompter_tokenizer = load_prompter()
 
-@app.get("/generate")
+@app.get("/")
 async def generate(text: str = Query(..., description="Input text to be processed by the model")):
     input_ids = prompter_tokenizer(text.strip() + " Rephrase:", return_tensors="pt").input_ids
     eos_id = prompter_tokenizer.eos_token_id
@@ -27,4 +25,4 @@ async def generate(text: str = Query(..., description="Input text to be processe
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=8000)