Update app.py

app.py

@@ -1,22 +1,13 @@
 import gradio as gr
 from transformers import pipeline
-import random
 
-# List of backend servers (simulated instances of the pipeline)
-backend_servers = [
-    pipeline("question-answering", model="google-bert/bert-large-uncased-whole-word-masking-finetuned-squad"),
-    pipeline("question-answering", model="google-bert/bert-large-uncased-whole-word-masking-finetuned-squad"),
-    pipeline("question-answering", model="google-bert/bert-large-uncased-whole-word-masking-finetuned-squad")
-]
+# Use a pipeline as a high-level helper
 
-# Function to select a backend server in a round-robin manner
-def select_backend():
-    return random.choice(backend_servers)
+pipe = pipeline("question-answering", model="Intel/dynamic_tinybert")
 
 # Define the function to answer questions
 def get_answer(text):
-    backend = select_backend()
-    result = backend(question=text, context=knowledge_base_text)
+    result = pipe(question=text, context=knowledge_base_text)
     return result["answer"]
 
 # Define the Gradio interface
@@ -60,7 +51,4 @@ iface = gr.Interface(
     theme='compact'
 )
 
-iface.launch()
-
-
-
+iface.launch()