added requirements.txt

app.py

@@ -1,4 +1,3 @@
-from json import load
 import os
 import torch
 import gradio as gr

model.py

@@ -1,12 +1,8 @@
-import math
-import logging
 import torch
 import torch.nn.functional as F
 from typing import Optional
 from torch import nn
 
-logger = logging.getLogger(__name__)
-
 
 class RMSNorm(nn.Module):
     def __init__(self, dim, eps):
@@ -314,7 +310,7 @@ class Transformer(nn.Module):
                    
         with torch.no_grad():
             for ret_seq in range(num_return_sequences):
-                logger.info(f"Sequence #{ret_seq + 1}:")
+                print(f"Sequence #{ret_seq + 1}:")
                 for _ in range(max_length - input_ids.shape[1]):
                     outputs = self(input_ids, use_cache=True)
                     next_token_logits = outputs[:, -1, :]

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+torch
+transformers