Update app.py

app.py

@@ -176,8 +176,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextGenerationPipe
 import torch
 
 # Load model and tokenizer
-model_name = "FreedomIntelligence/Apollo-7B"
+# model_name = "FreedomIntelligence/Apollo-7B"
 # model_name = "emilyalsentzer/Bio_ClinicalBERT"
+model_name = "FreedomIntelligence/Apollo-2B"
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)