Create app.py

app.py

@@ -0,0 +1,21 @@
+from litellm import completion
+
+# Configuration du modèle
+model_name = "gemini/gemini-1.5-pro"
+api_key = os.environ.get("GEMINI_API_KEY")
+
+# Exemple de fonction de base
+def get_completion(prompt):
+    try:
+        response = completion(
+            model=model_name, 
+            messages=[{"role": "user", "content": prompt}],
+            api_key=api_key
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Erreur: {str(e)}"
+
+# Pour les Spaces Hugging Face
+def predict(prompt):
+    return get_completion(prompt)