Added option for gemma and tinyllama

app.py

@@ -5,13 +5,22 @@ import time
 
 st.title("Transaction Summarizer")
 
+# Input for selecting model
+model_selection = st.radio("Select model to use:", ["Gemma", "Tinyllama"])
+
 # Input for submitting new transactions
 new_transactions = st.text_area("Enter your transactions (comma-separated)").split(',')
 new_transactions = [i.strip() for i in new_transactions if i.strip()]
 
+# Determine URL based on model selection
+if model_selection == "Gemma":
+    base_url = "https://api.runpod.ai/v2/cdvoxheesm31dc/"
+elif model_selection == "Tinyllama":
+    base_url = "https://api.runpod.ai/v2/0wnm75vx5o77s1/"
+
 # Submit button for new transactions
 if st.button("Submit New Transactions"):
-    url = "https://api.runpod.ai/v2/0wnm75vx5o77s1/run"
+    url = base_url + "run"
     
     # Retrieve API key from Streamlit secrets
     api_key = st.secrets["api_key"]
@@ -39,7 +48,7 @@ if st.button("Submit New Transactions"):
         st.write(f"New Job ID: {job_id}")
 
         # Keep checking status until it's no longer 'IN_QUEUE'
-        status_url = f"https://api.runpod.ai/v2/0wnm75vx5o77s1/status/{job_id}"
+        status_url = f"{base_url}status/{job_id}"
         status = "IN_QUEUE"
         while status != "COMPLETED":
             status_response = requests.get(status_url, headers=headers)