Spaces:
Sleeping
Sleeping
from flask import Flask, request, jsonify | |
from transformers import pipeline | |
app = Flask(__name__) | |
model = pipeline('text-generation', model='Erfan11/Neuracraft') | |
def generate_text(): | |
input_text = request.json.get('input_text', '') | |
result = model(input_text, max_length=100) | |
return jsonify(result) | |
if __name__ == '__main__': | |
app.run(debug=True, host='0.0.0.0', port=5000) |