Update app.py

app.py

@@ -43,17 +43,12 @@ def generate(file, prompt, temperature, max_new_tokens, top_p, repetition_penalt
             }
 
             try:
-                stream = client.text_generation(f"{prompt} Output the response in the following JSON format: {{'generated_sentence': 'The generated sentence text', 'confidence_score': 0.9}}", **generate_kwargs, stream=True, details=True, return_full_text=False)
+                stream = client.text_generation(f"{prompt} Output the response in the following JSON format: {{'generated_sentence': 'The generated sentence text', 'confidence_score': 0.9}} {sentence}", **generate_kwargs, stream=True, details=True, return_full_text=False)
                 output = ""
                 for response in stream:
                     output += response.token.text
 
-                try:
-                    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}})
+                data.append({"original_sentence": sentence, "generated_data": output})
 
             except Exception as e:
                 print(f"Error generating data for sentence '{sentence}': {e}")