Update app.py

app.py

@@ -222,9 +222,9 @@ class GenerationStrategy(ABC):
     def generate(self, generator: 'BaseGenerator', prompt: str, model_kwargs: Dict[str, Any], **kwargs) -> str:
         pass
 
-@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)