naonauno commited on
Commit
d836403
·
verified ·
1 Parent(s): 5618897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = AutoModelForSeq2SeqGeneration.from_pretrained(self.model_name)
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: