Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,10 +6,6 @@ import os
 # Read token from environment (set as secret in Space settings)
 token = os.environ.get("token")
 
-def infer(sent, max_length, num_return_sequences):
-    generator = pipeline('text-generation', model='gpt2')
-
-    return generator(sent, max_length=max_length, num_return_sequences=num_return_sequences)
 
 def instantiate_gpt2(max_length_ : int, num_return_sequences : int, text : str) -> dict:
     pipe = pipeline(task='text-generation', model='Iscte-Sintra/GPT2-Kriolu', tokenizer='Iscte-Sintra/GPT2-Kriolu', token=token)