Update: minor changes

app.py

@@ -4,6 +4,7 @@ from huggingface_hub import login
 import os
 from dotenv import load_dotenv
 load_dotenv() 
+
 HUGGINGFACE_TOKEN = os.environ.get("HUGGINGFACE_TOKEN")
 login(token=HUGGINGFACE_TOKEN)
 

src/brain.py

@@ -4,7 +4,6 @@ model_name = "google/gemma-2b"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
-
 def generate_answers(query):
     input_ids = tokenizer(query, return_tensors="pt")
     output = model.generate(**input_ids)