Hugging Face's logo

Spaces:
khalednabawi11
/
Med-Chatbot-API
Sleeping

App Files Community
khalednabawi11 commited on
Commit
495be45
·
verified ·
1 Parent(s): 93c5295

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ logging.basicConfig(level=logging.DEBUG)
29
  logger = logging.getLogger(__name__)
30
 
31
  # Load model and tokenizer
32
- model_name = "FreedomIntelligence/Apollo-7B"
33
  tokenizer = AutoTokenizer.from_pretrained(model_name)
34
  model = AutoModelForCausalLM.from_pretrained(model_name)
35
  tokenizer.pad_token = tokenizer.eos_token
 
29
  logger = logging.getLogger(__name__)
30
 
31
  # Load model and tokenizer
32
+ model_name = "FreedomIntelligence/Apollo-2B"
33
  tokenizer = AutoTokenizer.from_pretrained(model_name)
34
  model = AutoModelForCausalLM.from_pretrained(model_name)
35
  tokenizer.pad_token = tokenizer.eos_token