Upload 4 files

Dockerfile

@@ -0,0 +1,27 @@
+# Use the official Python 3.9 image
+FROM python:3.9
+
+# Set the working directory to /code
+WORKDIR /code
+
+# Copy the current directory contents into the container at /code
+COPY ./requirements.txt /code/requirements.txt
+
+# Install requirements.txt 
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+# Switch to the "user" user
+USER user
+# Set home to the user's home directory
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README (5).md

@@ -0,0 +1,11 @@
+---
+title: text-generation-GPT2
+emoji: 🌍
+colorFrom: blue
+colorTo: red
+sdk: docker
+pinned: false
+license: mit
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,52 @@
+# Install the necessary packages
+# pip install accelerate transformers fastapi pydantic torch
+
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+from pydantic import BaseModel
+from fastapi import FastAPI
+# Import the required library
+from transformers import pipeline
+
+# Initialize the FastAPI app
+app = FastAPI(docs_url="/")
+
+# Define the request model
+class RequestModel(BaseModel):
+    input: str
+
+# Define a greeting endpoint
+@app.get("/")
+def greet_json():
+    return {"message": "working..."}
+
+# Define the text generation endpoint
+@app.post("/generatetext")
+def get_response(request: RequestModel):
+    # Define the task and model
+    task = "text-generation"
+    model_name = "gpt2"
+
+    # Define the input text, maximum output length, and the number of return sequences
+    input_text = request.input
+    max_output_length = 50
+    num_of_return_sequences = 1
+
+    # Initialize the text generation pipeline
+    text_generator = pipeline(
+        task,
+        model=model_name
+    )
+
+    # Generate text sequences
+    generated_texts = text_generator(
+        input_text,
+        max_length=max_output_length,
+        num_return_sequences=num_of_return_sequences
+    )
+
+    # Extract and return the generated text
+    generated_text = generated_texts[0]['generated_text']
+    return {"generated_text": generated_text}
+
+# To run the FastAPI app, use the command: uvicorn <filename>:app --reload

requirements.txt

@@ -0,0 +1,6 @@
+fastapi==0.74.*
+requests==2.27.*
+uvicorn[standard]==0.17.*
+sentencepiece==0.1.*
+torch==1.11.*
+transformers==4.*