Spaces:
Sleeping
Sleeping
File size: 5,342 Bytes
31acbbd 75264c5 31acbbd 75264c5 31acbbd 6233258 31acbbd 75264c5 6233258 31acbbd 75264c5 31acbbd 75264c5 6233258 31acbbd 6233258 31acbbd 75264c5 31acbbd 75264c5 31acbbd 75264c5 31acbbd 75264c5 31acbbd 75264c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 |
import streamlit as st
import os
import numpy as np
import matplotlib.pyplot as plt
from ibm_watsonx_ai.foundation_models import Model
from matplotlib.colors import ListedColormap
# Function to get credentials from user input
def get_credentials():
url = "https://us-south.ml.cloud.ibm.com"
apikey = os.getenv("IBM_TOKEN")
if apikey:
return {"url": url, "apikey": apikey}
return None
# Function to modify the generated number according to the rules
def modify_number(number):
modified_number = ''
for digit in number:
if digit not in '01': # Change any digit that is not 0 or 1 to 1
modified_number += '1'
else:
modified_number += digit
return modified_number
# Function to create the grid based on the boolean string
def create_grid(boolean_string):
if len(boolean_string) != 16:
st.error("The generated string is not 16 digits long.")
return None
# Convert string to a list of integers
grid = np.array([int(bit) for bit in boolean_string]).reshape(4, 4)
# Create a colormap for the grid
cmap = ListedColormap(['white', 'black'])
# Plot the grid
fig, ax = plt.subplots(figsize=(2, 2)) # Shrink the graph size
ax.matshow(grid, cmap=cmap)
# Add grid lines and remove ticks
ax.set_xticks(np.arange(-0.5, 4, 1), minor=True)
ax.set_yticks(np.arange(-0.5, 4, 1), minor=True)
ax.grid(which='minor', color='gray', linestyle='-', linewidth=2)
ax.tick_params(left=False, bottom=False, labelleft=False, labelbottom=False)
return fig
# Function to interpolate between start and end values
def interpolate_value(start_value, end_value, position, max_position):
difference = end_value - start_value
interpolated_value = start_value + (difference * (position / max_position))
interpolated_value_int = int(interpolated_value)
binary_representation = bin(interpolated_value_int)[2:].zfill(16)
return binary_representation
# Streamlit app
def main():
st.title("Granite Star Search with IBM")
# Get credentials from the user
credentials = get_credentials()
if not credentials:
st.error("Please enter your API key.")
return
# Model and parameters
model_id = "ibm/granite-13b-chat-v2"
parameters = {
"decoding_method": "greedy",
"max_new_tokens": 900,
"repetition_penalty": 1.05
}
project_id = os.getenv("PROJECT_ID")
# Initialize the model
model = Model(
model_id=model_id,
params=parameters,
credentials=credentials,
project_id=project_id
)
# Sliders for selecting the start value and position
start_value = st.slider("Start Value", 0, 65535, 32768)
max_slider_value = 256
position = st.slider("Radial Out", 0, max_slider_value, 0)
# Define the distinct end values for each of the four numbers
end_values = [0, 65280, 65534, 255]
# Interpolate each value based on the slider position
interpolated_values = [
interpolate_value(start_value, end_values[0], position, max_slider_value), # Number 1
interpolate_value(start_value, end_values[1], position, max_slider_value), # Number 2
interpolate_value(start_value, end_values[2], position, max_slider_value), # Number 3
interpolate_value(start_value, end_values[3], position, max_slider_value) # Number 4
]
# Display the interpolated boolean strings
st.write("Interpolated Boolean Strings:")
for i, boolean_string in enumerate(interpolated_values):
st.write(f"Number {i+1}: {boolean_string}")
# Manual input for the 5th grid
manual_input = st.text_input("Enter a 16-digit boolean number (e.g., 1010101010101010):", "1010101010101010")
# Validate and modify the input if necessary
manual_input = modify_number(manual_input)
# Display the grids in a row, including the manual input
cols = st.columns(5) # Change the number of columns to 5
for i, boolean_string in enumerate(interpolated_values):
with cols[i]:
st.write(f"Grid for Number {i+1}:")
fig = create_grid(boolean_string)
if fig:
st.pyplot(fig)
# Add the 5th grid for the manual input
with cols[4]: # Use the 5th column for the manual input grid
st.write("Grid for Manual Input:")
if len(manual_input) != 16:
st.error("The input is not 16 digits long.")
else:
fig = create_grid(manual_input)
if fig:
st.pyplot(fig)
# Prepare the prompt for the model
prompt_input = """
Analyze the four 16 digit boolean numbers, identify the underlying pattern in the sequence, and use this underlying pattern to predict the next 16 digit boolean number. Return only the predicted boolean number, it must be 16 digits long with no explanation needed.
"""
formatted_question = " ".join(interpolated_values)
prompt = f"{prompt_input}\n{formatted_question}"
if st.button("Predict Next Number"):
generated_response = model.generate_text(prompt=prompt, guardrails=False)
st.write(f"**Predicted Next Boolean Number:** {generated_response}")
if __name__ == "__main__":
main()
|