Spaces:
Configuration error
Configuration error
from flask import Flask, request, jsonify | |
import pymongo | |
import requests | |
import json | |
from transformers import MT5ForConditionalGeneration, MT5Tokenizer | |
import torch | |
from torch.nn.functional import softmax | |
app = Flask(__name__) | |
#hazırladığım verilere ait database | |
mongo_client = pymongo.MongoClient('mongodb://localhost:27017/') | |
db = mongo_client['EgitimDatabase'] | |
collection = db['test'] | |
#kullanıcılara ait inputların kaydedileceği database | |
mongo_client = pymongo.MongoClient('mongodb://localhost:27017/') | |
db = mongo_client['EgitimDatabase'] | |
collection = db['input'] | |
outputs = model.generate(**inputs, output_scores=True, return_dict_in_generate=True, num_beams=5) | |
generated_text = tokenizer.decode(outputs.sequences[0], skip_special_tokens=True) | |
return generated_text | |
def predict(): | |
data = request.json | |
title = data.get('title') | |
keywords = data.get('keywords') | |
subheadings = data.get('subheadings') | |
# MongoDB'den ilgili verileri çekme | |
query = { | |
'title': title, | |
'keywords': {'$in': keywords.split(',')}, | |
'subheadings': {'$in': subheadings.split(',')} | |
} | |
documents = list(collection.find(query)) | |
if not documents: | |
return jsonify({'error': 'No documents found matching the query'}), 404 | |
# Verilerle metin oluşturma | |
generated_texts = [] | |
for doc in documents: | |
generated_text = generate_text(doc['title'], doc['keywords'], doc['subheadings']) | |
generated_texts.append(generated_text) | |
# Sonuçları döndürme | |
response = { | |
'title': title, | |
'keywords': keywords, | |
'subheadings': subheadings, | |
'generated_texts': generated_texts | |
} | |
return jsonify(response) | |
if __name__ == "__main__": | |
app.run(host='0.0.0.0', port=8080) | |