firsttry

Dockerfile

@@ -0,0 +1,18 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+RUN pip install --upgrade pip
+
+
+COPY requirements.txt .
+
+
+RUN pip install --no-cach-dir -r requirements.txt
+
+COPY . .
+
+EXPOSE 5000
+
+CMD ["python","app.py"]
+

app.py

@@ -0,0 +1,16 @@
+from transformers import pipeline
+from flask import Flask, request, jsonify
+
+
+
+app =Flask(__name__)
+generator=pipeline('text-generation',model='gbt2')
+@app.route('/generate',methods=['POST'])
+def generate():
+    data=request.json
+    prompt = data.get("prompt", "")
+    results = generator(prompt, max_length=50, num_return_sequences=1)
+    return jsonify(results)
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000)