Spaces:
Sleeping
added step 1-6 from semantic search
Browse files# Step 1 - Semantic Search
from sentence_transformers import SentenceTransformer
import torch
# Step 2 - Semantic Search
# Open the water_cycle.txt file in read mode with UTF-8 encoding
with open("water_cycle.txt", "r", encoding="utf-8") as file:
# Read the entire contents of the file and store it in a variable
water_cycle_text = file.read()
# Print the text below
print(water_cycle_text)
# Step 3 - Semantic Search
def preprocess_text(text):
# Strip extra whitespace from the beginning and the end of the text
cleaned_text = text.strip()
# Split the cleaned_text by every newline character (\n)
chunks = cleaned_text.split("\n")
# Create an empty list to store cleaned chunks
cleaned_chunks = []
# Write your for-in loop below to clean each chunk and add it to the cleaned_chunks list
for chunk in chunks:
stripped_chunk = chunk.strip()
if len(stripped_chunk) > 0:
cleaned_chunks.append(stripped_chunk)
# Print cleaned_chunks
print(cleaned_chunks)
# Print the length of cleaned_chunks
print(len(cleaned_chunks))
# Return the cleaned_chunks
return cleaned_chunks
# Step 4 - Semantic Search
# Load the pre-trained embedding model that converts text to vectors
model = SentenceTransformer('all-MiniLM-L6-v2')
def create_embeddings(text_chunks):
# Convert each text chunk into a vector embedding and store as a tensor
chunk_embeddings = model.encode(text_chunks, convert_to_tensor=True) # Replace ... with the text_chunks list
# Print the chunk embeddings
print(chunk_embeddings)
# Print the shape of chunk_embeddings
print(chunk_embeddings.shape)
# Return the chunk_embeddings
return chunk_embeddings
# Call the create_embeddings function and store the result in a new chunk_embeddings variable
chunk_embeddings = create_embeddings(cleaned_chunks) # Complete this line
# Call the preprocess_text function and store the result in a cleaned_chunks variable
#cleaned_chunks = preprocess_text(water_cycle_text) # Complete this line
# Step 5 - Semantic Search
def get_top_chunks(query, chunk_embeddings, text_chunks):
# Convert the query text into a vector embedding
query_embedding = model.encode(query, convert_to_tensor = True) # Complete this line
# Normalize the query embedding to unit length for accurate similarity comparison
query_embedding_normalized = query_embedding / query_embedding.norm()
# Normalize all chunk embeddings to unit length for consistent comparison
chunk_embeddings_normalized = chunk_embeddings / chunk_embeddings.norm(dim=1, keepdim=True)
# Calculate cosine similarity between query and all chunks using matrix multiplication
similarities = torch.matmul(chunk_embeddings_normalized, query_embedding_normalized) # Complete this line
# Print the similarities
print(similarities)
# Find the indices of the 3 chunks with highest similarity scores
top_indices = torch.topk(similarities, k=3).indices
# Print the top indices
print(top_indices)
# Create an empty list to store the most relevant chunks
top_chunks = []
# Loop through the top indices and retrieve the corresponding text chunks
for i in top_indices:
chunk = text_chunks[i]
top_chunks.append(chunk)
# Return the list of most relevant chunks
return top_chunks
# Step 6 - Semantic Search
# Call the get_top_chunks function with the original query
top_results = get_top_chunks("How does water get into the sky", chunk_embeddings, cleaned_chunks) # Complete this line
# Print the top results
print(top_results)
|
@@ -1,7 +1,120 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
import requests
|
| 3 |
from huggingface_hub import InferenceClient
|
|
|
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
SPOONACULAR_API_KEY = "71259036cfb3405aa5d49c1220a988c5"
|
| 6 |
|
| 7 |
def get_recipes(ingredient):
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import requests
|
| 3 |
from huggingface_hub import InferenceClient
|
| 4 |
+
#----------------------------------------------------------------------------------------------------------------------------------------------
|
| 5 |
|
| 6 |
+
# Step 1 - Semantic Search
|
| 7 |
+
from sentence_transformers import SentenceTransformer
|
| 8 |
+
import torch
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
# Step 2 - Semantic Search
|
| 13 |
+
# Open the water_cycle.txt file in read mode with UTF-8 encoding
|
| 14 |
+
with open("water_cycle.txt", "r", encoding="utf-8") as file:
|
| 15 |
+
# Read the entire contents of the file and store it in a variable
|
| 16 |
+
water_cycle_text = file.read()
|
| 17 |
+
# Print the text below
|
| 18 |
+
print(water_cycle_text)
|
| 19 |
+
|
| 20 |
+
|
| 21 |
+
|
| 22 |
+
# Step 3 - Semantic Search
|
| 23 |
+
def preprocess_text(text):
|
| 24 |
+
# Strip extra whitespace from the beginning and the end of the text
|
| 25 |
+
cleaned_text = text.strip()
|
| 26 |
+
|
| 27 |
+
# Split the cleaned_text by every newline character (\n)
|
| 28 |
+
chunks = cleaned_text.split("\n")
|
| 29 |
+
|
| 30 |
+
# Create an empty list to store cleaned chunks
|
| 31 |
+
cleaned_chunks = []
|
| 32 |
+
|
| 33 |
+
# Write your for-in loop below to clean each chunk and add it to the cleaned_chunks list
|
| 34 |
+
for chunk in chunks:
|
| 35 |
+
stripped_chunk = chunk.strip()
|
| 36 |
+
if len(stripped_chunk) > 0:
|
| 37 |
+
cleaned_chunks.append(stripped_chunk)
|
| 38 |
+
|
| 39 |
+
# Print cleaned_chunks
|
| 40 |
+
print(cleaned_chunks)
|
| 41 |
+
|
| 42 |
+
# Print the length of cleaned_chunks
|
| 43 |
+
print(len(cleaned_chunks))
|
| 44 |
+
|
| 45 |
+
# Return the cleaned_chunks
|
| 46 |
+
return cleaned_chunks
|
| 47 |
+
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
# Step 4 - Semantic Search
|
| 51 |
+
# Load the pre-trained embedding model that converts text to vectors
|
| 52 |
+
model = SentenceTransformer('all-MiniLM-L6-v2')
|
| 53 |
+
|
| 54 |
+
def create_embeddings(text_chunks):
|
| 55 |
+
# Convert each text chunk into a vector embedding and store as a tensor
|
| 56 |
+
chunk_embeddings = model.encode(text_chunks, convert_to_tensor=True) # Replace ... with the text_chunks list
|
| 57 |
+
|
| 58 |
+
# Print the chunk embeddings
|
| 59 |
+
print(chunk_embeddings)
|
| 60 |
+
|
| 61 |
+
# Print the shape of chunk_embeddings
|
| 62 |
+
print(chunk_embeddings.shape)
|
| 63 |
+
|
| 64 |
+
# Return the chunk_embeddings
|
| 65 |
+
return chunk_embeddings
|
| 66 |
+
|
| 67 |
+
# Call the create_embeddings function and store the result in a new chunk_embeddings variable
|
| 68 |
+
chunk_embeddings = create_embeddings(cleaned_chunks) # Complete this line
|
| 69 |
+
|
| 70 |
+
# Call the preprocess_text function and store the result in a cleaned_chunks variable
|
| 71 |
+
#cleaned_chunks = preprocess_text(water_cycle_text) # Complete this line
|
| 72 |
+
|
| 73 |
+
|
| 74 |
+
|
| 75 |
+
# Step 5 - Semantic Search
|
| 76 |
+
def get_top_chunks(query, chunk_embeddings, text_chunks):
|
| 77 |
+
# Convert the query text into a vector embedding
|
| 78 |
+
query_embedding = model.encode(query, convert_to_tensor = True) # Complete this line
|
| 79 |
+
|
| 80 |
+
# Normalize the query embedding to unit length for accurate similarity comparison
|
| 81 |
+
query_embedding_normalized = query_embedding / query_embedding.norm()
|
| 82 |
+
|
| 83 |
+
# Normalize all chunk embeddings to unit length for consistent comparison
|
| 84 |
+
chunk_embeddings_normalized = chunk_embeddings / chunk_embeddings.norm(dim=1, keepdim=True)
|
| 85 |
+
|
| 86 |
+
# Calculate cosine similarity between query and all chunks using matrix multiplication
|
| 87 |
+
similarities = torch.matmul(chunk_embeddings_normalized, query_embedding_normalized) # Complete this line
|
| 88 |
+
|
| 89 |
+
# Print the similarities
|
| 90 |
+
print(similarities)
|
| 91 |
+
|
| 92 |
+
# Find the indices of the 3 chunks with highest similarity scores
|
| 93 |
+
top_indices = torch.topk(similarities, k=3).indices
|
| 94 |
+
|
| 95 |
+
# Print the top indices
|
| 96 |
+
print(top_indices)
|
| 97 |
+
|
| 98 |
+
# Create an empty list to store the most relevant chunks
|
| 99 |
+
top_chunks = []
|
| 100 |
+
|
| 101 |
+
# Loop through the top indices and retrieve the corresponding text chunks
|
| 102 |
+
for i in top_indices:
|
| 103 |
+
chunk = text_chunks[i]
|
| 104 |
+
top_chunks.append(chunk)
|
| 105 |
+
|
| 106 |
+
# Return the list of most relevant chunks
|
| 107 |
+
return top_chunks
|
| 108 |
+
|
| 109 |
+
|
| 110 |
+
|
| 111 |
+
# Step 6 - Semantic Search
|
| 112 |
+
# Call the get_top_chunks function with the original query
|
| 113 |
+
top_results = get_top_chunks("How does water get into the sky", chunk_embeddings, cleaned_chunks) # Complete this line
|
| 114 |
+
# Print the top results
|
| 115 |
+
print(top_results)
|
| 116 |
+
|
| 117 |
+
#--------------------------------------------------------------------------------------------------------------------------------------------
|
| 118 |
SPOONACULAR_API_KEY = "71259036cfb3405aa5d49c1220a988c5"
|
| 119 |
|
| 120 |
def get_recipes(ingredient):
|