Upload 3 files

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.10-slim
+
+COPY . /app
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential git && \
+    pip install --no-cache-dir -r requirements.txt && \
+    apt-get clean && rm -rf /var/lib/apt/lists/*
+
+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 AutoModelForCausalLM, AutoTokenizer
+import torch
+
+app = FastAPI()
+
+model_name = "microsoft/phi-2"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+class MessageRequest(BaseModel):
+    username: str
+    message: str
+
+@app.post("/intro")
+def smart_intro(req: MessageRequest):
+    prompt = f"العميل اسمه {req.username}، وكتب: {req.message}\nرد عليه برد ذكي بسيط يحسسه إنك مهتم وهترد عليه بعد لحظات:"
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+    outputs = model.generate(**inputs, max_new_tokens=50, temperature=0.8)
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    reply = response.split(":")[-1].strip()
+    return {"reply": reply}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+torch
+transformers