cs2-ai-bot / app.py
Krepselis's picture
Update app.py
0169d92 verified
raw
history blame contribute delete
329 Bytes
import gradio as gr
from transformers import pipeline
model = pipeline("text2text-generation", model="google/flan-t5-large")
def generate_answer(prompt):
output = model(prompt, max_new_tokens=100)[0]["generated_text"]
return output
iface = gr.Interface(fn=generate_answer, inputs="text", outputs="text")
iface.launch()