ai01firebird commited on
Commit
9b23eab
·
verified ·
1 Parent(s): 8a0b589

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -16
app.py CHANGED
@@ -34,22 +34,21 @@ def text_to_emoji(text):
34
  #response = client.text_generation(prompt, max_new_tokens=50)
35
  #return response
36
 
37
-
38
- completion = client.chat.completions.create(
39
- model="mistralai/Mistral-Small-3.1-24B-Instruct-2503",
40
- messages=[
41
- {
42
- "role": "user",
43
- "content": [
44
- {
45
- "type": "text",
46
- "text": prompt
47
- }
48
- ]
49
- }
50
- ],
51
- max_tokens=25,
52
- )
53
  return completion.choices[0].message
54
 
55
  # Gradio UI
 
34
  #response = client.text_generation(prompt, max_new_tokens=50)
35
  #return response
36
 
37
+ completion = client.chat.completions.create(
38
+ model="mistralai/Mistral-Small-3.1-24B-Instruct-2503",
39
+ messages=[
40
+ {
41
+ "role": "user",
42
+ "content": [
43
+ {
44
+ "type": "text",
45
+ "text": prompt
46
+ }
47
+ ]
48
+ }
49
+ ],
50
+ max_tokens=25
51
+ )
 
52
  return completion.choices[0].message
53
 
54
  # Gradio UI