Update app.py

app.py

@@ -179,19 +179,12 @@ llm = Llama(model_path='./ggml-model-q8_0.gguf', n_ctx=2048)
 def read_email():
     df = pd.read_csv('./emails.csv')
     for i in df['Body']:
-        prompt_ = f"<s>system:{prompt}<|end|><s>user:{i}<|end|<s>assistant:"
+        prompt_ = f"<s>system:{prompt}<|end|><s>user:{i}<|end|><s>assistant:"
         output = llm(prompt_, max_tokens=1000, temperature=0.1)
+
         print("*"*50)
         t = output['choices'][0]['text']
-        if len(t)==0:
-            print('input : ',i)
-            print('output : ',t)
-            # t = t[t.find('json\n')+len('json\n'):]
-            extracted_details = json.loads(t)
-        else:
-            continue
-            
-        print(extracted_details)
+        extracted_details = json.loads(t[t.find('{'):t.find('}')+1])
         # print(extracted_details)
         print(type(extracted_details))
         meta_data = {