ProjectAnalysis / app.py
andreska's picture
Change the Space to become a service accepting input
99ec9bc verified
raw
history blame
303 Bytes
from transformers import pipeline
def analyze_project(project_data, question):
nlp = pipeline("text-generation", model="gpt2")
prompt = f"Analyze this project: {project_data}\n\nQuestion: {question}"
output = nlp(prompt, max_length=50, num_return_sequences=1)
return output[0]['generated_text']