Convert to ds_dict

app.py

@@ -69,16 +69,16 @@ def extract_json(gen_text, n_shot_learning=0):
     return gen_text[start_index:end_index]
 
 def score_model(model_uri, databricks_token, prompt):
-  dataset=pd.DataFrame({
+  ds_dict={
             "model": "debisoft/mpt-7b-awq-tester",
-            "prompt":[prompt],
-            "temperature": [0.5],
-            "max_tokens": [1000]})
+            "prompt": prompt,
+            "temperature": 0.5,
+            "max_tokens": 1000}
   headers = {
     "Authorization": f"Bearer {databricks_token}",
     "Content-Type": "application/json",
   }
-  ds_dict = {'dataframe_split': dataset.to_dict(orient='split')} if isinstance(dataset, pd.DataFrame) else create_tf_serving_json(dataset)
+  #ds_dict = {'dataframe_split': dataset.to_dict(orient='split')} if isinstance(dataset, pd.DataFrame) else create_tf_serving_json(dataset)
   data_json = json.dumps(ds_dict, allow_nan=True)
   print("***ds_dict: ")
   print(ds_dict)