Upload app.py
Browse files
app.py
CHANGED
@@ -123,9 +123,19 @@ def predict(user_input,company):
|
|
123 |
|
124 |
|
125 |
# Get response from the LLM
|
126 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
127 |
model_output = predict(user_input,company)
|
128 |
-
|
129 |
|
130 |
# While the prediction is made, log both the inputs and outputs to a local log file
|
131 |
# While writing to the log file, ensure that the commit scheduler is locked to avoid parallel
|
|
|
123 |
|
124 |
|
125 |
# Get response from the LLM
|
126 |
+
def predict_output (user_input,company):
|
127 |
+
if company == "AWS":
|
128 |
+
company = "AWS"
|
129 |
+
elif company == "IBM":
|
130 |
+
company = "IBM"
|
131 |
+
elif company =+ "Google":
|
132 |
+
company = "Google"
|
133 |
+
elif company == "Msft":
|
134 |
+
company = "Msft"
|
135 |
+
elif company == "Meta":
|
136 |
+
company = "Meta"
|
137 |
model_output = predict(user_input,company)
|
138 |
+
return model_output
|
139 |
|
140 |
# While the prediction is made, log both the inputs and outputs to a local log file
|
141 |
# While writing to the log file, ensure that the commit scheduler is locked to avoid parallel
|