Update handler.py

handler.py

@@ -1,46 +1,3 @@
-Hugging Face's logo
-Hugging Face
-Search models, datasets, users...
-Models
-Datasets
-Spaces
-Docs
-Solutions
-Pricing
-
-
-
-
-ammarnasr
-/
-CodeGen2_1B_merged 
-
-like
-0
-Text Generation
-Transformers
-PyTorch
-codegen
-custom_code
-Inference Endpoints
-Model card
-Files and versions
-Community
-Settings
-CodeGen2_1B_merged
-/
-handler.py
-ammarnasr's picture
-ammarnasr
-Update handler.py
-5f48ffc
-7 minutes ago
-raw
-history
-blame
-edit
-delete
-1.06 kB
 from typing import Any, Dict, List
 import torch
 import transformers
@@ -66,3 +23,6 @@ class EndpointHandler:
         generated_ids = self.model.generate(input_ids, **config)
         generated_text = self.tokenizer.decode(generated_ids[0], skip_special_tokens=True)
         return [{"generated_text": generated_text}]
+        
+            
+