Update app.py

app.py

@@ -1,22 +1,11 @@
-import pandas as pd
-import gradio as gr
 from transformers import pipeline
 
-emails_df = pd.read_csv('emails.csv', nrows=500, on_bad_lines='skip')
-context = " ".join(emails_df['message'].apply(lambda x: x.strip() if isinstance(x, str) else ''))
+model_path = '/content/model_output'
+text_gen = pipeline("text-generation", model=model_path, tokenizer=model_path)
 
-qa_pipeline = pipeline("question-answering")
+def answer_question(question):
+    result = text_gen(question, max_length=100, num_return_sequences=1)
+    return result[0]['generated_text']
 
-def answer_query(question):
-    try:
-        result = qa_pipeline(question=question, context=context)
-        return result['answer']
-    except Exception as e:
-        return str(e)  
-
-iface = gr.Interface(
-    fn=answer_query,
-    inputs=gr.Textbox(label="Enter your question:"),
-    outputs=gr.Textbox(label="Answer:")
-)
-iface.launch()
+iface = gr.Interface(fn=answer_question, inputs="text", outputs="text")
+iface.launch()