Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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("
|
29 |
-
llama_model = AutoModelForCausalLM.from_pretrained("
|
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)
|