CalvinBackup/Dioscuri
1
0
Fork 0
mirror of https://github.com/elder-plinius/Dioscuri.git synced 2026-02-12 17:22:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update dioscuri.py

Browse Source
This commit is contained in:
pliny
2024-04-02 21:52:16 -07:00
committed by GitHub
parent 6a6ee0ae3b
commit dae279f42e
1 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dioscuri.py
View File

@@ -34,15 +34,12 @@ def main():
# Temperature slider
temperature = st.slider("Temperature", 0.0, 1.0, 0.5, 0.01)
# Model dropdown
model_options = ["gemini-pro"]
selected_model = st.selectbox("Select Model", model_options, index=3) # Default to "gemini"
# Toggle for appending encoded string (jailbreak)
append_encoded_string = st.checkbox("Toggle Jailbreak", value=False)
if user_input:
prompt = user_input
if append_encoded_string:
# Append encoded string to the user input
prompt = user_input + " " + get_appended_prompt()
@@ -53,10 +50,7 @@ def main():
while retry_count < max_retries:
# Send the prompt to GeminiChat
if selected_model == "gemini-pro":
response = gemini_chat.send_message(prompt, temperature=temperature)
else:
response = gemini_chat.send_message(prompt, temperature=temperature, model=selected_model)
response = gemini_chat.send_message(prompt, temperature=temperature)
if response is not None:
break