Update app.py

app.py

@@ -166,18 +166,20 @@ def analyze_with_cpu_model(raw_outputs):
 
 # === MAIN GENERATION FUNCTION ===
 def generate_prediction_ensemble(scenario_text, max_len, temperature):
-    # Hardcoded testing mode
+    if not scenario_text.strip():
+        return "Please enter a scenario", "", ""
+
+    prompt = format_input(scenario_text)
+
     raw_outputs = [
-        '''Model 1:
-Based on the situation, predict potential hazards and injuries. During routine maintenance, a hydraulic press malfunctioned due to worn seals. The operator's hand was caught when the press unexpectedly activated.
-{"Cause of Accident": "Shock caused by Misjudgment, Hazardous Situation", "Degree of Injury": "High", "Hazards": ["Caught in or between caused by Pinch Point Action", "ELEVATED WORK PLATFORM", "POWERED INDUSTRIAL VEHICLE"]}'''
+        generate_single_model_output(model, tokenizer, prompt, max_length=max_len, temperature=temperature)
+        for model, tokenizer in zip(models, tokenizers)
     ]
 
     cause, injury = analyze_with_cpu_model(raw_outputs)
     raw_output_text = "\n\n".join([f"Model {i+1}:\n{resp}" for i, resp in enumerate(raw_outputs)])
     return cause, injury, raw_output_text
 
-
 # === FULL GRADIO INTERFACE ===
 def create_interface():
     with gr.Blocks(title="Multi-Model Safety Risk Predictor") as interface: