Yoxas commited on
Commit
49dab71
·
verified ·
1 Parent(s): 02403f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ index.add(embeddings)
25
  sentence_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', device=device)
26
 
27
  # Load the LLaMA model for response generation
28
- llama_tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
29
- llama_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf").to(device)
30
 
31
  # Load the summarization model
32
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn", device=0 if device == 'cuda' else -1)
 
25
  sentence_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', device=device)
26
 
27
  # Load the LLaMA model for response generation
28
+ llama_tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
29
+ llama_model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2").to(device)
30
 
31
  # Load the summarization model
32
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn", device=0 if device == 'cuda' else -1)