updated

app.py

@@ -64,4 +64,16 @@ async def root():
 @app.post("/generate/")
 async def generate_text(request: GenerateRequest):
     response = generate_text_from_model(request.prompt)
-    return {"response": response}
+    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