Update app.py

app.py

@@ -253,6 +253,7 @@ class GenerationStrategy(ABC):
 
 @observe()
 class DefaultStrategy(GenerationStrategy):
+    @observe()
     def generate(self, generator: 'BaseGenerator', prompt: str, model_kwargs: Dict[str, Any], **kwargs) -> str:
         input_ids = generator.tokenizer(prompt, return_tensors="pt").input_ids.to(generator.device)
         output = generator.model.generate(input_ids, **model_kwargs)