Update app.py

app.py

@@ -1,38 +1,22 @@
 import streamlit as st
 import openai
-from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
 
 # Configuration de l'API OpenAI
 openai.api_key = 'VOTRE_CLE_API_OPENAI'
 
-# Chargement du modèle et du tokenizer de Hugging Face
-model_name = "meta-llama/Meta-Llama-3.1-405B-Instruct"  # Vous pouvez remplacer par un autre modèle de Hugging Face si besoin
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-
 # Fonction pour générer une réponse à partir du modèle OpenAI
 def generate_openai_response(prompt):
     response = openai.Completion.create(
-        model="text-davinci-003",  # Assurez-vous d'utiliser le modèle correct ici
+        model="text-davinci-003",  # Vous pouvez remplacer par un autre modèle d'OpenAI si besoin
         prompt=prompt,
         max_tokens=150,
         temperature=0.7
     )
     return response.choices[0].text.strip()
 
-# Fonction pour générer une réponse à partir du modèle Hugging Face
-def generate_huggingface_response(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(inputs.input_ids, max_length=150, num_return_sequences=1)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
-
 # Interface utilisateur avec Streamlit
-st.title("Chatbot avec OpenAI et Hugging Face")
-st.write("Ce chatbot utilise les modèles de OpenAI et Hugging Face pour répondre à vos questions.")
-
-# Sélection du modèle à utiliser
-model_option = st.selectbox("Choisissez le modèle à utiliser:", ("OpenAI GPT-4", "Hugging Face GPT-2"))
+st.title("Chatbot avec OpenAI")
+st.write("Ce chatbot utilise le modèle OpenAI pour répondre à vos questions.")
 
 # Entrée utilisateur
 user_input = st.text_area("Vous:", "")
@@ -40,10 +24,7 @@ user_input = st.text_area("Vous:", "")
 if st.button("Envoyer"):
     if user_input:
         with st.spinner("Génération de la réponse..."):
-            if model_option == "OpenAI GPT-3":
-                response = generate_openai_response(user_input)
-            else:
-                response = generate_huggingface_response(user_input)
+            response = generate_openai_response(user_input)
         st.write("Chatbot:", response)
     else:
         st.warning("Veuillez entrer une question ou un message.")