Update app.py

app.py

@@ -26,11 +26,11 @@ async def generateTextUsingLLama(inputSettings:InputForGeneration) -> GeneratedO
     try:       
         output = text_generation.generateText(inputSettings)
         if output is not None:
-            return GeneratedOutput(status_code=200, generated_text=output)
+            return output
         else:
-            return GeneratedOutput(status_code=400, message="error when generating text")
+            return "error when generating text"
     except Exception as e:
-        return GeneratedOutput(status_code=500, message=str(e))
+        return str(e)
     
 if __name__ == "__main__":
     uvicorn.run(app=app)