Upload 9 files

.env

@@ -0,0 +1 @@
+HF_HOME=./models

.gitignore

@@ -0,0 +1,2 @@
+__pycache__
+node_modules

Dockerfile

@@ -0,0 +1,16 @@
+FROM python:3.11
+
+WORKDIR /api
+
+COPY ./ /api/
+
+RUN pip install --upgrade pip
+RUN pip install poetry
+RUN poetry install --no-root
+
+RUN mkdir /api/cache
+RUN chmod a+rwx /api/cache
+
+EXPOSE 7860
+
+CMD ["python", "main.py"]

README.md

@@ -1,10 +1,11 @@
 ---
-title: Img Gen
-emoji: 🦀
-colorFrom: blue
-colorTo: yellow
+title: Micapi
+emoji: 🚀
+colorFrom: red
+colorTo: red
 sdk: docker
 pinned: false
+app_file: main.py
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

main.py

@@ -0,0 +1,33 @@
+import os
+from transformers import CLIPProcessor, CLIPModel
+import torch
+from PIL import Image
+
+# Get the directory of the script
+script_directory = os.path.dirname(os.path.realpath(__file__))
+# Specify the directory where the cache will be stored (same folder as the script)
+cache_directory = os.path.join(script_directory, "cache")
+# Create the cache directory if it doesn't exist
+os.makedirs(cache_directory, exist_ok=True)
+
+# Load the CLIP processor and model
+clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32", cache_dir=cache_directory)
+clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32", cache_dir=cache_directory)
+
+# Text description to generate image
+text = "a cat sitting on a table"
+
+# Tokenize text and get features
+inputs = clip_processor(text, return_tensors="pt", padding=True)
+
+with torch.no_grad():
+  image_features = clip_model.get_image_features(**inputs)
+
+# Generate image from features
+generated_image = clip_model.generate_images(image_features)
+
+# Save the generated image
+output_image_path = "generated_image.png"
+Image.fromarray(generated_image).save(output_image_path)
+
+print("Image generated and saved as:", output_image_path)

mainHistory.py

@@ -0,0 +1,46 @@
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+from pydantic import BaseModel
+from fastapi import FastAPI
+
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+model_name = "facebook/blenderbot-1B-distill"
+
+# https://huggingface.co/models?sort=trending&search=facebook%2Fblenderbot
+# facebook/blenderbot-3B
+# facebook/blenderbot-1B-distill
+# facebook/blenderbot-400M-distill
+# facebook/blenderbot-90M
+# facebook/blenderbot_small-90M
+
+# https://www.youtube.com/watch?v=irjYqV6EebU
+
+app = FastAPI()
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+class req(BaseModel):
+  prompt: str
+
+@app.get("/")
+def read_root():
+  return FileResponse(path="templates/index.html", media_type="text/html")
+
+@app.post("/api")
+def read_root(data: req):
+  print("Prompt:", data.prompt)
+
+  input_text = data.prompt
+  
+  # Tokenize the input text
+  input_ids = tokenizer.encode(input_text, return_tensors="pt")
+  
+  # Generate output using the model
+  output_ids = model.generate(input_ids, num_beams=5, no_repeat_ngram_size=2)
+  generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
+  
+  answer_data = { "answer": generated_text }
+  print("Answer:", generated_text)
+  
+  return answer_data

models.md

@@ -0,0 +1,7 @@
+# Model list
+* [microsoft/DialoGPT-small](https://huggingface.co/microsoft/DialoGPT-small)
+* [microsoft/DialoGPT-medium](https://huggingface.co/microsoft/DialoGPT-medium)
+* [microsoft/DialoGPT-large](https://huggingface.co/microsoft/DialoGPT-large)
+
+# Download locations
+* Github Codespaces: /home/codespace/.local/lib/python3.10/site-packages/transformers/models/

pyproject.toml

@@ -0,0 +1,17 @@
+[tool.poetry]
+name = "img gen"
+version = "0.0.1"
+description = "A project to test image generation with AI models"
+authors = ["CubeBeveled <[email protected]>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.11"
+transformers = "^4.39.1"
+torch = "^2.2.1"
+pillow = "^10.2.0"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

requirements.txt

@@ -0,0 +1,4 @@
+transformers
+torch
+poetry
+pillow