Update app.py

app.py

@@ -1,21 +1,21 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the question-answering model
-model = pipeline("question-answering", model="Rishitha0208/new-llm-for-advanced-materials")
+# Load the model
+model = pipeline("text-generation", 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']
+def generate_response(prompt):
+    # Generate text from the model
+    response = model(prompt, max_length=150, num_return_sequences=1)
+    return response[0]['generated_text']
 
 # Define the Gradio interface
 interface = gr.Interface(
-    fn=answer_question,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your question here...", label="Question"),
+    fn=generate_response,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt"),
     outputs="text",
     title="Polymer Knowledge Model",
-    description="A model fine-tuned to answer questions about polymers."
+    description="A model fine-tuned for generating text related to polymers."
 )
 
 # Launch the interface