5

app.py

@@ -1,16 +1,26 @@
 from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 from transformers import pipeline
 
 # Create a new FastAPI app instance
 app = FastAPI()
 
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allow all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allow all methods
+    allow_headers=["*"],  # Allow all headers
+)
+
 # Initialize the text generation pipeline
 pipe = pipeline("text2text-generation", model="google/flan-t5-small")
 
 # Define a root path to verify the server is running
 @app.get("/")
 def read_root():
-    return {"message": "Welcome to the FastAPI application. Use /generate to generate text 2."}
+    return {"message": "Welcome to the FastAPI application. Use /generate to generate text."}
 
 # Define a function to handle the GET request at `/generate`
 @app.get("/generate")