variux

package biojava.biojava_project;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.LinkedHashMap;

variux

from transformers import pipeline

with open("/home/arteaga/prot/scripts/input_denovo_multiple.txt", "r") as f:
    for line in f:
        secuencia = []
        protgpt2 = pipeline('text-generation', model="/home/arteaga/prot/data/finetuned")
        sequences = protgpt2(line, max_length=50, min_length = 0, do_sample=True, top_k=950, repetition_penalty=1.2, num_return_sequences=10, eos_token_id=0)
        with open("/home/arteaga/prot/scripts/output_denovo_multiple.txt", "w") as wf:
            for seq in sequences:
                wf.write(seq)