Update app.py

app.py

@@ -1,34 +1,20 @@
 import gradio as gr
-
-# Sample classes
-class ResultA:
-    def __init__(self, query):
-        self.response = f"Result from Function A for query: {query}"
-
-class ResultB:
-    def __init__(self, query):
-        self.response = f"Result from Function B for query: {query}"
-
-class ResultC:
-    def __init__(self, query):
-        self.response = f"Result from Function C for query: {query}"
+from gen_ai import traditional_model,llm_model
 
 # Functions that return class instances
 def function_a(query):
-    return ResultA(query)
+    return traditional_model(query).predict()
 
 def function_b(query):
-    return ResultB(query)
+    return llm_model(query)
 
-def function_c(query):
-    return ResultC(query)
 
 # Function to handle user input
 def handle_query(function_choice, query):
     function_map = {
-        "Function A": function_a,
+        "FinBERT": function_a,
         "Function B": function_b,
-        "Function C": function_c,
+
     }
     
     if function_choice in function_map: