Spaces:

yakine
/

Dataset_Generator

Runtime error

yakine commited on Aug 9, 2024

Commit

f6e79d6

verified ·

1 Parent(s): 1ff09ca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 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