dict

app.py

@@ -22,16 +22,16 @@ def extract_json(gen_text, n_shot_learning=0):
 
 def score_model(model_uri, databricks_token, prompt):
   prompt = "Korea is a"
-  dataset=pd.DataFrame({
+  ds_dict={
             "model": "debisoft/mpt-7b-awq-tester",
             "prompt": prompt,
             "temperature": 0.5,
-            "max_tokens": 1000})
+            "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)