Create app.py

app.py

@@ -0,0 +1,22 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import pipeline, set_seed
+
+app = FastAPI()
+
+# Load GPT-2 model
+generator = pipeline("text-generation", model="gpt2")
+set_seed(42)
+
+class PromptRequest(BaseModel):
+    prompt: str
+    max_length: int = 50
+
+@app.get("/")
+def read_root():
+    return {"message": "Ku soo dhawoow GPT-2 Somali Chat API"}
+
+@app.post("/generate/")
+def generate_text(request: PromptRequest):
+    result = generator(request.prompt, max_length=request.max_length, num_return_sequences=1)
+    return {"generated_text": result[0]['generated_text']}