| from fastapi import FastAPI |
| from transformers import pipeline |
| import uvicorn |
|
|
| app = FastAPI() |
|
|
| print("Caricamento modello amogaddy/GenerAI...") |
| generator = pipeline("text-generation", model="amogaddy/GenerAI") |
|
|
| @app.get("/") |
| def health_check(): |
| return {"status": "running", "model": "GenerAI"} |
|
|
| @app.post("/ask") |
| def ask(prompt: str): |
| output = generator(prompt, max_length=100, num_return_sequences=1) |
| return {"result": output[0]['generated_text']} |
|
|
| if __name__ == "__main__": |
| uvicorn.run(app, host="0.0.0.0", port=7860) |
|
|
