Merge branch 'main' of https://github.com/mattsoh/LLM

mattsoh · Sep 20, 2024 · 498a5c5 · 498a5c5
2 parents 6cf5f50 + 28653ef
commit 498a5c5
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 2 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/Procfile b/Procfile
@@ -0,0 +1 @@
+web: python generate.py
diff --git a/generate.py b/generate.py
@@ -1,9 +1,12 @@
+from flask import Flask, request, render_template_string
 import torch
 import tiktoken
 import gdown
 import os
 from helper import generate_and_print_sample, GPT
 
+app = Flask(__name__)
+
 GPT_CONFIG = {
     "vocab_size": 50257,
     "context_length": 256,
@@ -16,7 +19,6 @@
 
 FILE_ID = "1mkhifEI6HQoiVahnGgZaRYsVU23req_M"
 MODEL_PATH = "model.pth"
-
 if not os.path.exists(MODEL_PATH):
     try:
         print("Downloading model.pth from Google Drive...")

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,3 @@
 tiktoken==0.7.0
 torch==2.4.1
-gdown
+gdown
diff --git a/runtime.txt b/runtime.txt
@@ -0,0 +1 @@
+python-3.12.0