Spaces:

VelaTest
/

PredictionHub

Running

Vela commited on Feb 15

Commit

60d1a5c

1 Parent(s): a0f1fda

added function to return predict

Files changed (2) hide show

src/api/__pycache__/main.cpython-312.pyc CHANGED Viewed

Binary files a/src/api/__pycache__/main.cpython-312.pyc and b/src/api/__pycache__/main.cpython-312.pyc differ

src/api/main.py CHANGED Viewed

@@ -24,6 +24,6 @@ def display_dimention(message : str = "Hello World"):
 def display_prediction(message : str = "Give me a sms to predict"):
     try:
         prediction = encoding_model.get_prediction(message)
-        return prediction
     except Exception as e:
         return f"Unable to fetch the data {e}"

 def display_prediction(message : str = "Give me a sms to predict"):
     try:
         prediction = encoding_model.get_prediction(message)
+        return {"message" : f"Given sms is a {prediction}"}
     except Exception as e:
         return f"Unable to fetch the data {e}"