Upload app.py

app.py

@@ -123,9 +123,19 @@ def predict(user_input,company):
 
 
     # Get response from the LLM
-
+def predict_output (user_input,company):
+    if company == "AWS":
+       company = "AWS"
+    elif company == "IBM":
+       company = "IBM"
+    elif company =+ "Google":
+       company = "Google"
+    elif company == "Msft":
+       company = "Msft"
+    elif company == "Meta":
+       company = "Meta"
     model_output = predict(user_input,company)
-
+    return model_output
 
     # While the prediction is made, log both the inputs and outputs to a local log file
     # While writing to the log file, ensure that the commit scheduler is locked to avoid parallel