updated

app.py

@@ -64,16 +64,5 @@ async def root():
 @app.post("/generate/")
 async def generate_text(request: GenerateRequest):
     response = generate_text_from_model(request.prompt)
-    removing="<s>[INST] \n<|user|>\nCould you please suggest some new startup companies that have entered the market from 2018 and might be good for investment? Provide only name of the company.\n<|assistant|> [/INST] Cred\n</s>"
-    rev=removing.split(" ")
-    text1=response.split(" ")
-    for i in rev:
-        try:
-            text1.remove(i)
-        except:
-            pass
-    final=""
-    for i in text1:
-        final+=i+" "
-        final+="\n"
-    return final
+    return {"response": response}
+