k

app.py

@@ -21,12 +21,12 @@ def extract_json(gen_text, n_shot_learning=0):
     return gen_text[start_index:end_index]
 
 def score_model(model_uri, databricks_token, prompt):
-  prompt = "Seoul is a"
+  prompt = "Korea is a"
   dataset=pd.DataFrame({
             "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",