Update app.py

app.py

@@ -1,7 +1,6 @@
 import json
 from huggingface_hub import InferenceClient
 import gradio as gr
-import PyPDF2
 import random
 import pandas as pd
 from io import BytesIO
@@ -50,10 +49,11 @@ def generate(file, prompt, temperature, max_new_tokens, top_p, repetition_penalt
                     output += response.token.text
 
                 try:
-                    json_output = json.loads(output)
-                    data.append({"original_sentence": sentence, "generated_data": json_output})
+                    generated_data = json.loads(output)
+                    data.append({"original_sentence": sentence, "generated_data": generated_data})
                 except json.JSONDecodeError:
                     print(f"Error decoding JSON for sentence '{sentence}': {output}")
+                    data.append({"original_sentence": sentence, "generated_data": {"generated_sentence": output, "confidence_score": None}})
 
             except Exception as e:
                 print(f"Error generating data for sentence '{sentence}': {e}")