Update app.py

app.py

@@ -61,6 +61,8 @@ model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True
 )
 
+config_json = model.config.to_dict()
+
 def format_model_info(config):
     info = []
     important_keys = [
@@ -69,9 +71,12 @@ def format_model_info(config):
     ]
     for key in important_keys:
         if key in config:
-            info.append(f"**{key}:** {config[key]}")
+            value = config[key]
+            # Convert torch_dtype to string representation if it exists
+            if key == "torch_dtype" and hasattr(value, "name"):
+                value = value.name
+            info.append(f"**{key}:** {value}")
     return "\n".join(info)
-
 @spaces.GPU
 def generate_response(system_prompt, user_prompt, temperature, max_new_tokens, top_p, repetition_penalty, top_k):
     # Construct the full prompt with system and user messages