Update app.py

app.py

@@ -22,6 +22,7 @@ api_token = "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"  # Replace with your actual
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     low_cpu_mem_usage=True,
+    return_dict=True,
 
     revision="main",  # Or the desired revision
     token=api_token