gruhit-patel
commited on
Commit
•
a1025a5
1
Parent(s):
7152f53
Modified repo name
Browse files- QuoteGenerator.py +1 -1
QuoteGenerator.py
CHANGED
@@ -3,7 +3,7 @@ from transformers.pipelines import TextGenerationPipeline
|
|
3 |
from typing import Union
|
4 |
|
5 |
class QuoteGenerator():
|
6 |
-
def __init__(self, model_name:str='
|
7 |
self.model_name = model_name
|
8 |
self.quote_generator: TextGenerationPipeline
|
9 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
|
|
3 |
from typing import Union
|
4 |
|
5 |
class QuoteGenerator():
|
6 |
+
def __init__(self, model_name:str='gruhit-patel/quote-generator-v2'):
|
7 |
self.model_name = model_name
|
8 |
self.quote_generator: TextGenerationPipeline
|
9 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|