Spaces:

Erfan11
/

Textwizai

Running

Erfan11 commited on 1 day ago

Commit

ff60146

•

1 Parent(s): d76f6e3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,53 +1,34 @@
-import os
-import requests
-import yaml
 from flask import Flask, request, jsonify
-from dotenv import load_dotenv
-from PIL import Image
-import io
 from transformers import pipeline
-# Load environment variables
-load_dotenv()
-api_key = os.getenv('HF_API_KEY')
-model_path = os.getenv('MODEL_PATH')
 app = Flask(__name__)
-# Load configuration
-with open('config.yaml', 'r') as file:
-    config = yaml.safe_load(file)
-def get_model_predictions(text):
-    headers = {"Authorization": f"Bearer {api_key}"}
-    payload = {"inputs": text}
-    response = requests.post(f"https://api-inference.huggingface.co/models/{model_path}", headers=headers, json=payload)
-    return response.json()
-def process_image(image_file):
-    image = Image.open(image_file)
-    # Implement image processing here
-    return "Image processed"
-@app.route('/predict', methods=['POST'])
-def predict():
-    data = request.get_json()
-    text = data.get('text')
-    image_file = data.get('image')
-    if text:
-        prediction = get_model_predictions(text)
-        return jsonify(prediction)
-    elif image_file:
-        image = io.BytesIO(image_file)
-        result = process_image(image)
-        return jsonify({"result": result})
-    else:
-        return jsonify({"error": "No input provided"})
-@app.route('/', methods=['GET'])
-def index():
-    return "Welcome to My AI! Use /predict to interact."
 if __name__ == '__main__':
-    app.run(debug=True, use_reloader=False)

 from flask import Flask, request, jsonify
 from transformers import pipeline
 app = Flask(__name__)
+# Initialize the text generation pipeline
+generator = pipeline('text-generation', model='gpt2')  # Replace 'gpt2' with your model
+@app.route('/')
+def home():
+    return '''
+    <html>
+    <body>
+        <h1>Text Generation</h1>
+        <form action="/generate" method="post">
+            <textarea name="text_input" rows="4" cols="50" placeholder="Enter text here..."></textarea><br>
+            <input type="submit" value="Generate">
+        </form>
+    </body>
+    </html>
+    '''
+@app.route('/generate', methods=['POST'])
+def generate():
+    text_input = request.form['text_input']
+    if text_input:
+        # Generate text using the model
+        result = generator(text_input, max_length=100, num_return_sequences=1)
+        generated_text = result[0]['generated_text']
+        return jsonify({'generated_text': generated_text})
+    return jsonify({'error': 'No input provided'}), 400
 if __name__ == '__main__':
+    app.run(debug=True)