Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ class MissionGenerator:
|
|
25 |
# Using FLAN-T5-base, a free and lightweight model good for instruction following
|
26 |
self.model_name = "google/flan-t5-base"
|
27 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
28 |
-
self.model =
|
29 |
self.context = MissionContext()
|
30 |
|
31 |
def format_conversation_history(self) -> str:
|
|
|
25 |
# Using FLAN-T5-base, a free and lightweight model good for instruction following
|
26 |
self.model_name = "google/flan-t5-base"
|
27 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
28 |
+
self.model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
|
29 |
self.context = MissionContext()
|
30 |
|
31 |
def format_conversation_history(self) -> str:
|