else

app.py

@@ -12,7 +12,10 @@ databricks_token  = os.getenv('TENATCH_TOKEN')
 model_uri  = "http://15.152.197.215/v1/completions"
 
 def extract_json(gen_text, n_shot_learning=0):
-    start_index = gen_text.index("### Response:\n{") + 14
+	if(n_shot_learning == -1) :
+		start_index = 0
+	else :
+    	start_index = gen_text.index("### Response:\n{") + 14
     if(n_shot_learning > 0) :
         for i in range(0, n_shot_learning):
             gen_text = gen_text[start_index:]
@@ -122,7 +125,7 @@ Give a score for the product. Format your response as a JSON object with \
 	#return json.dumps(extract_json(gen_text, 3))
 	gen_text = response["choices"][0]["text"]
 	#return gen_text
-	return extract_json(gen_text, 0)
+	return extract_json(gen_text, -1)
 	#return json.dumps(response)
 
 #iface = gr.Interface(fn=greet, inputs="text", outputs="text")