Commit History
-
Create README.md
Verified signatureimport requests import pandas as pd from transformers import BioGptTokenizer, BioGptForCausalLM def get_uniprot_data(protein_list): """Fetches UniProt data for a list of proteins using the UniProt API.""" base_url = "https://rest.uniprot.org/uniprotkb/search?query=" fields = "accession,id,protein_name,gene_names,organism,comment(FUNCTION)" results = [] for protein in protein_list: query_url = f"{base_url}{protein}&fields={fields}&format=tsv" response = requests.get(query_url) if response.status_code == 200: data = response.text.split("\n") if len(data) > 1: results.append(data[1].split("\t")) # Extract first result row else: print(f"Failed to retrieve {protein}: {response.status_code}") return results def analyze_protein_candidates(dataset_path, sheet_name): """Analyzes proteins from dataset and retrieves UniProt data.""" df = pd.read_excel(dataset_p…238ea0b