init space

.gitignore

@@ -0,0 +1,2 @@
+# files
+*.DS_Store

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY app.py .
+
+# Expose port 7860 (Spaces yêu cầu)
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,23 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+# Khởi tạo app FastAPI
+app = FastAPI()
+
+# Load model và tokenizer
+model_name = "bmd1905/BARTpho2-ViT5-text2text"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+
+# Định nghĩa schema cho request
+class InputText(BaseModel):
+    text: str
+
+# Endpoint POST /generate
+@app.post("/generate")
+def generate_text(item: InputText):
+    inputs = tokenizer(item.text, return_tensors="pt", truncation=True, max_length=512)
+    outputs = model.generate(**inputs, max_new_tokens=128)
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return {"answer": response}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+transformers==4.40.0
+torch>=1.13.1