Trying something else

app.py

@@ -1,20 +1,24 @@
 import gradio as gr
-# from huggingface_hub import InferenceClient
-import random
 from transformers import pipeline
 
-def random_response(message, history):
-    return random.choice(["Yes", "No"])
+# Load the question answering model
+question_answerer = pipeline("question-answering", model="mbwolff/distilbert-base-uncased-finetuned-squad")
 
-def eqa(message, history):
-    if len(history) >= 1 and history[-1]['role'] == 'assistant' and history[-1]['content'] == 'Okay':
-        pipe = pipeline(model="mbwolff/distilbert-base-uncased-finetuned-squad")
-        return pipe(question=message, context=history[-2]['content'])['answer']
-    else:
-        return 'Okay'
+def answer_question(question, context):
+    result = question_answerer(question=question, context=context)
+    return result["answer"]
 
-gr.ChatInterface(
-#    fn=random_response,
-    fn=eqa,
-    type="messages"
-).launch()
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=answer_question,
+    inputs=[
+        gr.inputs.Textbox(lines=7, placeholder="Enter your question here..."),
+        gr.inputs.Textbox(lines=7, placeholder="Enter the context here..."),
+    ],
+    outputs="text",
+    title="Question Answering Chatbot",
+    description="Ask questions about the provided context."
+)
+
+# Launch the interface
+iface.launch()

appOLD.py

@@ -0,0 +1,20 @@
+import gradio as gr
+# from huggingface_hub import InferenceClient
+import random
+from transformers import pipeline
+
+def random_response(message, history):
+    return random.choice(["Yes", "No"])
+
+def eqa(message, history):
+    if len(history) >= 1 and history[-1]['role'] == 'assistant' and history[-1]['content'] == 'Okay':
+        pipe = pipeline(model="mbwolff/distilbert-base-uncased-finetuned-squad")
+        return pipe(question=message, context=history[-2]['content'])['answer']
+    else:
+        return 'Okay'
+
+gr.ChatInterface(
+#    fn=random_response,
+    fn=eqa,
+    type="messages"
+).launch()