polymers / app.py
rish13's picture
Update app.py
8937d91 verified
raw
history blame
726 Bytes
import gradio as gr
from transformers import pipeline
# Load the question-answering model
model = pipeline("question-answering", model="Rishitha0208/new-llm-for-advanced-materials")
def answer_question(question):
# Generate an answer from the model
result = model(question=question, context="") # Provide an empty context if not required
return result['answer']
# Define the Gradio interface
interface = gr.Interface(
fn=answer_question,
inputs=gr.Textbox(lines=2, placeholder="Enter your question here...", label="Question"),
outputs="text",
title="Polymer Knowledge Model",
description="A model fine-tuned to answer questions about polymers."
)
# Launch the interface
interface.launch()