update

distilgpt2/Dockerfile

@@ -1,10 +1,6 @@
 # please visit https://github.com/xfyun/aiges/releases to get stable and suitable iamges.
 
-FROM docker.io/library/python:3.8.9
-
-
-RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
-RUN sed -i 's|security.debian.org/debian-security|mirrors.ustc.edu.cn/debian-security|g' /etc/apt/sources.list
+FROM public.ecr.aws/iflytek-open/cuda-go-python-base:11.6-1.17-3.9.13-ubuntu1804
 
 
 WORKDIR /home/user/app

distilgpt2/app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import pipeline, set_seed
 
+
 def inference(text):
     model_path = "distilgpt2"
     generator = pipeline('text-generation', model=model_path)
@@ -13,13 +14,6 @@ def inference(text):
 
     return output
 
-    # tokenizer = GPT2Tokenizer.from_pretrained(model_path)
-    # model = GPT2Model.from_pretrained(model_path)
-    # encoded_input = tokenizer(text, return_tensors='pt')
-    # output = model(**encoded_input)
-    # print(output)
-    # return output
-
 
 examples = [["Hello, I’m a language model."]]