Update app.py

app.py

@@ -4,6 +4,8 @@ import requests
 from transformers import GPT2LMHeadModel, GPT2Tokenizer, LlamaTokenizer, LlamaForCausalLM, pipeline
 from huggingface_hub import HfFolder, login
 from io import StringIO
+import os
+from flask import Flask, request, jsonify
 
 # Load GPT-2 model and tokenizer