add model to output

app.py

@@ -10,7 +10,7 @@ def fetch_images(postID):
         response.raise_for_status()  # This will raise an exception for HTTP errors
         data = response.json()
         items = data["items"]
-        table_data = [[item["id"], item["meta"]["prompt"], item["meta"].get("negativePrompt", "N/A"), item["meta"]["sampler"], item["meta"]["steps"], item["meta"]["cfgScale"], item["width"], item["height"], item["url"]] for item in items]
+        table_data = [[item["id"], item["meta"]["prompt"], item["meta"].get("negativePrompt", "N/A"), item["meta"]["Model"],item["meta"]["sampler"], item["meta"]["steps"], item["meta"]["cfgScale"], item["width"], item["height"], item["url"]] for item in items]
         return table_data
     except requests.RequestException as e:
         st.error(f"Failed to fetch images: {e}")
@@ -26,6 +26,7 @@ def display_images(df):
             with col2:
                 st.markdown(f"**Prompt:** {row['prompt']}")
                 st.markdown(f"**Negative Prompt:** {row['negativePrompt']}")
+                st.markdown(f"**Model:** {row['Model']}")
                 st.markdown(f"**Sampler:** {row['sampler']}    |    **Steps:** {row['steps']}")
                 st.markdown(f"**Config Scale:** {row['cfgScale']}")
                 st.markdown(f"**Image Size:** {row['height']} H x {row['width']} W")
@@ -51,7 +52,7 @@ def main():
         # Convert postID to integer if necessary
         try:
             postID_int = int(postID_input)
-            df = pd.DataFrame(fetch_images(postID_int), columns=["id", "prompt", "negativePrompt", "sampler", "steps", "cfgScale", "width", "height", "url"])
+            df = pd.DataFrame(fetch_images(postID_int), columns=["id", "prompt", "negativePrompt", "Model", "sampler", "steps", "cfgScale", "width", "height", "url"])
             if not df.empty:
                 display_images(df)
             else: