Fix Dockerfile formatting and remove unused code

Dockerfile

@@ -10,17 +10,17 @@ 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
+# # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
-# Switch to the "user" user
+# # Switch to the "user" user
 USER user
-# Set home to the user's home directory
+# # 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
+# # 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 the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
 # Start the FastAPI app on port 7860, the default port expected by Spaces

app.py

@@ -1,23 +1,11 @@
 from fastapi import FastAPI
-from transformers import pipeline
-from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 # Create a new FastAPI app instance
 app = FastAPI(docs_url="/", redoc_url="/new_redoc")
 
-# Create a translation pipeline
-tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
-model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small")
-
 
 # Create a POST endpoint
 @app.get("/generate")
 def generate(text: str):
     """Traduce del castellano al inglés"""
-
-    input_text = "translate Spanish to English: " + text
-    input_ids = tokenizer(input_text, return_tensors="pt").input_ids
-
-    outputs = model.generate(input_ids)
-    print(tokenizer.decode(outputs[0]))
-    return {"output": tokenizer.decode(outputs[0])}
+    return {"output": text}

docker-compose.yml

@@ -0,0 +1,13 @@
+version: '3.8'
+services:
+  your-service:
+    image: prueba-ia
+    ports:
+      - "7860:7860"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [ gpu ]

requirements.txt

@@ -3,4 +3,4 @@ requests==2.27.*
 uvicorn[standard]==0.17.*
 sentencepiece==0.1.*
 torch==1.11.*
-transformers==4.*
+#transformers==4.*