Update app.py

app.py

@@ -8,14 +8,17 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
 def generate_json(prompt):
-    instruction = f"Generate a JSON object from the following description:\n{prompt}"
+    # Ensure the prompt clearly asks for JSON output
+    instruction = f"Generate a JSON object with the following properties: {prompt}"
     result = generator(instruction, max_length=256, do_sample=False)
     generated_text = result[0]["generated_text"]
-
+    
     try:
+        # Try to parse the text as JSON
         parsed = json.loads(generated_text)
         formatted_json = json.dumps(parsed, indent=2)
     except Exception as e:
+        # Return raw text if parsing fails
         formatted_json = f"Raw Output:\n{generated_text}\n\nError parsing JSON: {e}"
 
     return formatted_json