Update app.py

app.py

@@ -1,25 +1,33 @@
-import streamlit as st
-import requests
+from flask import Flask, request, jsonify, render_template
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Fonction pour interagir avec le modèle Hugging Face
+app = Flask(__name__)
+
+# Charger le modèle et le tokenizer depuis Hugging Face
+model_name = "salmapm/llama2_salma"
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# Définir une fonction de génération de texte
 def generate_text(prompt):
-    API_URL = "https://api-inference.huggingface.co/models/salmapm/llama2_salma"
-    payload = {"inputs": prompt}
-    try:
-        response = requests.post(API_URL, json=payload)
-        response.raise_for_status()  # Cela déclenchera une exception pour les codes d'état HTTP 4xx/5xx
-        data = response.json()
-        return data[0]['generated_text']
-    except requests.exceptions.HTTPError as http_err:
-        return f"Erreur HTTP : {http_err}"
-    except requests.exceptions.RequestException as e:
-        return f"Erreur de requête : {e}"
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(**inputs, max_length=50, num_return_sequences=1, temperature=0.7)
+    generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return generated_text
 
-st.title("chatfinance")
-prompt = st.text_input("Entrez un texte:")
-if st.button("Générer"):
+@app.route('/')
+def home():
+    return render_template('index.html')
+
+@app.route('/generate', methods=['POST'])
+def generate():
+    data = request.json
+    prompt = data.get('prompt', '')
     if prompt:
-        result = generate_text(prompt)
-        st.write(result)
+        generated_text = generate_text(prompt)
+        return jsonify({"generated_text": generated_text})
     else:
-        st.write("Veuillez entrer un texte pour générer la suite.")
+        return jsonify({"error": "Veuillez entrer du texte."}), 400
+
+if __name__ == '__main__':
+    app.run(debug=True)