d

app.py

@@ -56,7 +56,7 @@ I am building an online community to help people to find dates.
 {{"solution": "FindDates.com", "problem": "finding a date", "features": "online community to help people find dates", "target_customer": "people looking for a date", "fg_will_use": "True", "reason_to_use": "I am looking for an online community to help people find dates. FindDates.com meets my needs and I would use it to find my next great date.","fg_will_pay": "True", "reason_to_pay": "I would not pay for it as I am looking for an online community to help people find dates. But for products related to dating, paying for it would be a no-brainer.","fg_will_invest": "False", "reason_to_invest": "There are many online dating platforms already.","score": "40"}}
 """
 
-def extract_json(gen_text, response, n_shot_learning=0):
+def extract_json(gen_text, n_shot_learning=0):
     if(n_shot_learning == -1) :
         start_index = 0
     else :
@@ -99,6 +99,7 @@ def score_model(model_uri, databricks_token, prompt):
       print(response.status_code)
       print(response.txt)
       print("End***No choices!!!")
+
   return response_json
 
 def get_completion(prompt):
@@ -133,7 +134,7 @@ Give a score for the product. Format your response as a JSON object with \
 	print(total_prompt)
 	response = get_completion(total_prompt)
 	gen_text = response["choices"][0]["text"]
-	return json.dumps(extract_json(gen_text, response, -1))
+	return json.dumps(extract_json(gen_text, -1))
 
 #iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 #iface.launch()