DemahAlmutairi commited on
Commit
fca7c1b
·
verified ·
1 Parent(s): 30de0dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
-
4
  # Project 1: Bilingual Story Generation
5
  english_story_generator = pipeline("text-generation", model="gpt2")
6
  arabic_story_generator = pipeline("text-generation", model="aubmindlab/arabert")
 
1
  import gradio as gr
2
  from transformers import pipeline
3
+ import torch
4
  # Project 1: Bilingual Story Generation
5
  english_story_generator = pipeline("text-generation", model="gpt2")
6
  arabic_story_generator = pipeline("text-generation", model="aubmindlab/arabert")