Update app.py
Browse files
app.py
CHANGED
@@ -71,8 +71,8 @@ class DocumentRetrievalAndGeneration:
|
|
71 |
return generate_text
|
72 |
def initialize_llm2(self,model_id):
|
73 |
|
74 |
-
tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-1b3"
|
75 |
-
model = BloomForCausalLM.from_pretrained("bigscience/bloom-1b3"
|
76 |
result_length = 200
|
77 |
inputs = tokenizer(prompt, return_tensors="pt")
|
78 |
# return generate_text
|
|
|
71 |
return generate_text
|
72 |
def initialize_llm2(self,model_id):
|
73 |
|
74 |
+
tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-1b3")#, local_files_only=True)
|
75 |
+
model = BloomForCausalLM.from_pretrained("bigscience/bloom-1b3")#, local_files_only=True)
|
76 |
result_length = 200
|
77 |
inputs = tokenizer(prompt, return_tensors="pt")
|
78 |
# return generate_text
|